Pandas - 透视表转换为方阵

3

I have this simple dataframe in a data.csv file:

I,C,v
a,b,1
b,a,2
e,a,1
e,c,0
b,d,1
a,e,1
b,f,0

我想对它进行透视,然后返回一个正方形表格(作为矩阵)。到目前为止,我已经读取了数据框并使用以下代码构建了一个透视表:

df = pd.read_csv('data.csv')
d = pd.pivot_table(df,index='I',columns='C',values='v')
d.fillna(0,inplace=True)

正确获取:

C  a  b  c  d  e  f
I                  
a  0  1  0  0  1  0
b  2  0  0  1  0  0
e  1  0  0  0  0  0

现在我想返回一个方形表格,其中包含行中缺失的列索引,使得结果表格如下:

C  a  b  c  d  e  f
I                  
a  0  1  0  0  1  0
b  2  0  0  1  0  0
c  0  0  0  0  0  0
d  0  0  0  0  0  0
e  1  0  0  0  0  0
f  0  0  0  0  0  0
1个回答

7

reindex可以添加行和列,并填充缺失的值为0:

index = d.index.union(d.columns)
d = d.reindex(index=index, columns=index, fill_value=0)

产量
   a  b  c  d  e  f
a  0  1  0  0  1  0
b  2  0  0  1  0  0
c  0  0  0  0  0  0
d  0  0  0  0  0  0
e  1  0  0  0  0  0
f  0  0  0  0  0  0

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接