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