python-3.x - 如何将虚拟变量列转换为多列？

Question

我知道如何将行拆分为列，但如何处理以下 dataframe？

<表类="s-表"> <头> <日>日期 虚拟 <日>平均 标签 <正文> 1-19 1 20 l1 1-19 0 40 l1 1-27 1 100 l2 1-27 0 140 l2

预期的数据帧:

<表类="s-表"> <头> <日>日期 avg_t avg_c 标签 <正文> 1-19 20 40 l1 1-27 100 140 l2

当dummy 等于1 时，avg 为20，并重命名为avg_t 作为列。类似于 avg_c 列。

我试过:

df.groupby(['dummy','avg']).size().unstack

但是，它不起作用。

Answer 1

您还可以使用 janitor 中的 pivot_wider:.

# pip install pyjanitor
import pandas as pd
import janitor

df.pivot_wider(index = ['date', 'lable'], names_from = 'dummy')

Out[19]: 
   date lable  avg_0  avg_1
0  1-19    l1     40     20
1  1-27    l2    140    100

如果你想要 t 和 c:

(df.assign(dummy = df['dummy'].map({0:'c', 1:'t'})).
    pivot_wider(index = ['date', 'lable'], names_from = 'dummy'))

Out[20]: 
   date lable  avg_c  avg_t
0  1-19    l1     40     20
1  1-27    l2    140    100