在Pandas身上,我如何将一组行展平

2024-06-26 03:15:31 发布

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

我对python中的熊猫还不太熟悉,如果有任何帮助,我将不胜感激。我一直在谷歌上下搜索,但似乎无法破解它。在

例如,我有一个包含6列的csv文件。我正在尝试将行分组,以便将每行的所有数据展平为一行。在

所以如果我的数据是这样的:

event event_date event_time name height age 1 2015-05-06 14:00 J Bloggs 185 24 1 2015-05-06 14:00 P Smith 176 55 1 2015-05-06 14:00 T Kirk 193 22 2 2015-05-14 17:00 B Gates 178 72 2 2015-05-14 17:00 J Mayer 184 42

最后我想让它变成这样

event  event_date   event_time    name_1     height_1   age_1   name_2     height_2   age_2    name_3    height_3   age_3                                          
1      2015-05-06   14:00         J Bloggs   185        24      P Smith    176        55       T Kirk    193        22                                         
2      2015-05-14   17:00         B Gates    178        72      J Mayer    184        42
                                                                                                                                                                           .           

正如您在上面看到的,前3行中的第一个事件已被展平为一行,列展开以容纳行数据。第二个事件已被展平,列中填充了数据。

任何帮助都将得到评估。在


Tags: 文件csv数据nameeventagedatetime
2条回答

步骤:

1)计算Groupby对象的累计计数。加1,使标题按照所需的DF格式化。在

2)设置与索引轴相同的分组列以及计算的cumcounts,然后unstack。另外,根据最底层对标题进行排序。在

3)重命名多索引列并相应地展平以获得单个标题。在


cc = df.groupby(['event','event_date','event_time']).cumcount() + 1
df = df.set_index(['event','event_date','event_time', cc]).unstack().sort_index(1, level=1)
df.columns = ['_'.join(map(str,i)) for i in df.columns]
df.reset_index()

enter image description here

你用一张长桌子做了一张宽桌子。通常在数据分析中,你想做的恰恰相反。这里有一个方法,它首先统计每个变量名、身高和年龄的出现次数,然后按你想要的方式旋转它们。在

df['group_num'] = df.groupby(['event', 'event_date','event_time']).cumcount() + 1
df = df.sort_values('group_num')
df1 = df.set_index(['event', 'event_date','event_time', 'group_num']).stack().reset_index()
df1['var_names'] = df1['level_4'] + '_' + df1['group_num'].astype(str)
df1 = df1.drop(['group_num', 'level_4'], axis=1)
df1.set_index(['event', 'event_date', 'event_time', 'var_names']).squeeze().unstack('var_names')

var_names                   age_1 age_2 age_3 height_1 height_2 height_3  \
event event_date event_time                                                
1     2015-05-06 14:00         24    55    22      185      176      193   
2     2015-05-14 17:00         72    42  None      178      184     None   

var_names                      name_1   name_2  name_3  
event event_date event_time                             
1     2015-05-06 14:00       J Bloggs  P Smith  T Kirk  
2     2015-05-14 17:00        B Gates  J Mayer    None  

相关问题 更多 >