python - 如何使用 pandas 取消数据透视

Question

例如，我有一个像这样的数据框

df = {'name':['Jennifer','Vivian','Trisha'],
     'married':[1,1,0],
     'Mon': [0, 0,1],
     'Tu':[1,0,0],
     'Wed':[0,1,0]}

如何将虚拟变量融合到这样的一列中:

我尝试使用 pd.melt() 但它只是将几列堆叠为一列并更改列的长度。有人可以帮我解决这个问题吗？预先感谢您!

Answer 1

这应该有效:

df.replace({'married':{1:'Married', 0: 'Single'}}). \
   melt(id_vars=['married', 'name'], var_name='Workday'). \
   query('value == 1'). \
   drop('value', axis=1)

#    married      name Workday
# 2   Single    Trisha     Mon
# 3  Married  Jennifer      Tu
# 7  Married    Vivian     Wed