根據熊貓中各行的唯一值創建新列

Question

我正在嘗試在pandas列中使用唯一值來生成一組新列。 這是一個示例DataFrame ：

    meas1  meas2  side  newindex
0       1      3     L         0
1       2      4     R         0
2       6      8     L         1
3       7      9     R         1

我想將我的測量列與關鍵列“相乘”以生成如下所示的DafaFrame ：

   meas1_L  meas1_R  meas2_L  meas2_R
0        1        2        3        4
1        6        7        8        9

請注意，它本質上是這個問題的反面 。

Answer 1

使用DataFrame.pivot ：

# Perform the pivot.
df = df.pivot(index='newindex', columns='side').rename_axis(None)

# Format the columns.
df.columns = df.columns.map('_'.join)

結果輸出：

   meas1_L  meas1_R  meas2_L  meas2_R
0        1        2        3        4
1        6        7        8        9

Answer 2

使用groupby.prod另一種解決方案：

df = df.groupby(['side', 'newindex']).prod().unstack(level=0)
df.columns = ['_'.join(c[0::]) for c in df.columns]

          meas1_L  meas1_R  meas2_L  meas2_R
newindex                                    
0               1        2        3        4
1               6        7        8        9