Pandas数据框:如何按列中的值进行分组,并从分组值中创建新列

时间:2016-01-01 13:10:37

标签: python pandas dataframe

我有一个包含两列的数据框:

x y
0 1
1 1
2 2
0 5
1 6
2 8
0 1
1 8
2 4
0 1
1 7
2 3

我想要的是:

x val1 val2 val3 val4
0 1 5 1 1
1 1 6 8 7
2 2 8 4 3

我知道x列中的值会重复N次。

1 个答案:

答案 0 :(得分:5)

您可以使用groupby/cumcount分配列号,然后拨打pivot

import pandas as pd

df = pd.DataFrame({'x': [0, 1, 2, 0, 1, 2, 0, 1, 2, 0, 1, 2],
                   'y': [1, 1, 2, 5, 6, 8, 1, 8, 4, 1, 7, 3]})

df['columns'] = df.groupby('x')['y'].cumcount()
#     x  y  columns
# 0   0  1        0
# 1   1  1        0
# 2   2  2        0
# 3   0  5        1
# 4   1  6        1
# 5   2  8        1
# 6   0  1        2
# 7   1  8        2
# 8   2  4        2
# 9   0  1        3
# 10  1  7        3
# 11  2  3        3

result = df.pivot(index='x', columns='columns')
print(result)

产量

         y         
columns  0  1  2  3
x                  
0        1  5  1  1
1        1  6  8  7
2        2  8  4  3

或者,如果你真的可以依赖x重复中的值 N次,

N = 3
result = pd.DataFrame(df['y'].values.reshape(-1, N).T)

产量

   0  1  2  3
0  1  5  1  1
1  1  6  8  7
2  2  8  4  3

使用reshape比调用groupby/cumcountpivot更快,但它更快 不太健壮,因为它依赖于y中出现的正确顺序的值。