我尝试使用group by来创建新的数据帧,但我需要多索引才能保持一致。无论子类别是否存在,我都希望它像以下一样创建:
import pandas as pd
df = pd.DataFrame(
{'Cat 1':['A','A','A','B','B','B','B','C','C','C','C','C','D'],
'Cat 2':['A','B','A','B','B','B','A','B','B','B','B','B','A'],
'Num': [1,1,1,1,1,1,1,1,1,1,1,1,1]})
print df.groupby(['Cat 1','Cat 2']).sum()
输出如下:
Num
Cat 1 Cat 2
A A 2
B 1
B A 1
B 3
C B 5
D A 1
但我希望它看起来像
Num
Cat 1 Cat 2
A A 2
B 1
B A 1
B 3
C A Nan
B 5
D A 1
B Nan
我读取了不同的数据,然后以这种格式添加了一列,因此生成的数组看起来像:
Num Num_added_later
Cat 1 Cat 2
A A 2 12
B 1 5
B A 1 5
B 3 3
C A Nan 5
B 5 5
D A 1 1
B Nan 3
答案 0 :(得分:7)
您可以根据两个 Cat 列和 reindex 创建新索引:
import pandas as pd
new_index = pd.MultiIndex.from_product([df["Cat 1"].unique(), df["Cat 2"].unique()], names = ["Cat 1", "Cat 2"])
df.groupby(['Cat 1','Cat 2']).sum().reindex(new_index)
答案 1 :(得分:4)