Pandas:用重复条目重塑数据帧

2024-03-29 10:10:24 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个名为df

    deathtype    height    deaths
0   AMS           4900       1
1   AMS           5150       1
2   AMS           5300       1
3   Avalanche     5350       14
4   Avalanche     5600       4
5   Avalanche     5700       1
6   Avalanche     5800       17
7   Unexplained   8500       1
8   Unexplained   8560       1

我正试图将数据重塑为以下内容:

deaths         1                4          14       17
deathtype               
AMS           4900,5150,5300    0          0        0
Avalanche     5700              5600       5350     5800
Unexplained   8500, 8560        0          0        0

我知道pivot表不能实现这一点,因为aggfunc对重复值使用平均值,这意味着对于1的所有deaths值,都将记录平均值。一张透视表给了我以下信息

df.pivot_table(index='deathtype', columns='deaths', values='height', fill_value='0')

deaths           1              4      14     17
deathtype               
AMS           5116.666667       0      0      0
Avalanche     5700.000000       5600   5350   5800
Unexplained   8530.000000       0      0      0

我在找一些关于如何做这件事的建议。看起来pivot表不是正确的方法。有人能给点建议吗。你知道吗


Tags: 数据df记录建议平均值pivotamsheight
1条回答
网友
1楼 · 发布于 2024-03-29 10:10:24

groupby与aggregate by join一起使用,然后按unstack重塑形状:

d = lambda x: ', '.join(x.astype(str))
df = df.groupby(['deathtype', 'deaths'])['height'].agg(d).unstack(fill_value='0')
print (df)
deaths                     1     4     14    17
deathtype                                      
AMS          4900, 5150, 5300     0     0     0
Avalanche                5700  5600  5350  5800
Unexplained        8500, 8560     0     0     0

细节

print (df.groupby(['deathtype', 'deaths'])['height'].agg(lambda x: ', '.join(x.astype(str))))
deathtype    deaths
AMS          1         4900, 5150, 5300
Avalanche    1                     5700
             4                     5600
             14                    5350
             17                    5800
Unexplained  1               8500, 8560
Name: height, dtype: object

另一个带有^{}的解决方案:

df = df.pivot_table(index='deathtype', 
                    columns='deaths', 
                    values='height', 
                    fill_value='0', 
                    aggfunc=lambda x: ', '.join(x.astype(str)))

相关问题 更多 >