python中pandas數據幀的矩陣表示

Question

我有一個像數據框架

 from   to  Amt
 a      b   100
 a      c   200
 a      d   220
 b      a   250
 b      c   300
 b      d   330
 c      a   100
 c      b   120
 c      d   320
 d      a   211
 d      b   980
 d      c   430    

我想以矩陣格式表示它

     a     b     c    d
a    0    100    200  220
b   250    0     300  330
c   100   120    0    320
d   211   980    430   0

怎么實現那個..

我已經將打印列表作為表格數據鏈接。但沒有得到我想要的。

Answer 1

您需要透視數據。 這是一個例子。

pivot_df = df.pivot(index='from', columns='to', values='Amt')

對於事前進行小數計算，您可以使用groupby()然后transform('sum') 。 它類似於SQL窗口函數和。

df['sums'] =  df.groupby('from')['amt'].transform('sum')
df['frac'] = df['amt'] / df['sums']
df.pivot(index='from', columns='to', values='frac')

Answer 2

您需要透視數據框。 請參見http://pandas.pydata.org/pandas-docs/stable/reshaping.html

df.pivot(index="from", columns="to",values="Amt" )

Answer 3

您也可以使用pivot_table實現：

df_pivoted = pd.pivot_table(df, index='from', columns='to', fill_value=0)
print(df_pivoted)

      Amt               
to      a    b    c    d
from                    
a       0  100  200  220
b     250    0  300  330
c     100  120    0  320
d     211  980  430    0