Python Pandas 數據框將 col 的值重塑為新的 col

pisto 发表于 Dev

手槍

我有一個看起來像這樣的數據框：

    col1    col2    col3    col4    col5    col6    
0   1.1     a       29      b       c       d
1   2.3     a       29      b       c       d
2   10.3    a       29      b       c       d
3   6.5     a       29      b       c       d
4   34.7    a       29      b       c       d

5   6.3     e       25      f       g       h
6   7.1     e       25      f       g       h
7   36.0    e       25      f       g       h
8   74.2    e       25      f       g       h
9   64.7    e       25      f       g       h

我想重塑它，使其看起來像：

    col1    col2    col3    col4    col5    col6    col7    col8    col9    col10
0   a       29      b       c       d       1.1     2.3     10.3    6.5     34.7
1   e       25      f       g       h       6.3     7.1     36.0    74.2    64.7

我試過了

df = df.groupby('col2')['col1'].apply(list)

col2 
a    [1.1, 2.3, 10.3, 6.5, 34.7]
b    [6.3, 7.1, 36.0, 74.2, 64.7]

但是我不知道如何從列表中獲取每個值到一個新的列。

耶斯列

使用GroupBy.cumcount有DataFrame.pivot：

df['new'] = df.groupby(['col2','col3','col4','col5','col6']).cumcount()

df = df.pivot(index=['col2','col3','col4','col5','col6'], columns='new', values='col1').reset_index()
df.columns = [f'col{i + 1}' for i in range(len(df.columns))]
print (df)
  col1  col2 col3 col4 col5  col6  col7  col8  col9  col10
0    a    29    b    c    d   1.1   2.3  10.3   6.5   34.7
1    e    25    f    g    h   6.3   7.1  36.0  74.2   64.7

本文收集自互联网，转载请注明来源。

如有侵权，请联系 [email protected] 删除。