R语言:将DataFrame按Auto聚合,ClasPri转为列的宽表转换求助
解决方法:将长格式DataFrame转换为宽格式
没问题!这其实是个典型的**数据透视(pivot)**场景,用Pandas就能快速实现你想要的格式转换。我给你两种常用的方法,你可以根据需求选择:
方法一:使用pivot方法
这是最直接的方式,完美适配你的数据(每个Auto+ClasPri组合唯一):
import pandas as pd # 假设你的原始DataFrame名为df df_pivoted = df.pivot(index='Auto', columns='ClasPri', values='Total').reset_index() # 移除默认的列名层级,让格式更整洁 df_pivoted.columns.name = None # 确保列顺序是你需要的0、1、2、3(可选,如果原始ClasPri顺序已经是这个的话可以省略) df_pivoted = df_pivoted[['Auto', 0, 1, 2, 3]]
代码说明:
pivot(index='Auto', columns='ClasPri', values='Total'):把Auto作为行索引,ClasPri的唯一值作为新列,Total的值自动填充到对应位置reset_index():将Auto从索引变回普通列,完全匹配你期望的格式- 最后调整列顺序,保证输出的列严格按照0到3的顺序排列
方法二:使用pivot_table方法
如果你的数据后续可能出现重复的Auto+ClasPri组合(虽然你说当前每个组合唯一,但这个方法更通用),可以用pivot_table并指定聚合函数:
import pandas as pd df_pivoted = df.pivot_table( index='Auto', columns='ClasPri', values='Total', aggfunc='first' # 因为每个组合唯一,用first/sum/mean结果都一致 ).reset_index() df_pivoted.columns.name = None df_pivoted = df_pivoted[['Auto', 0, 1, 2, 3]]
额外提示
如果你的ClasPri列不是整数类型(比如是字符串格式的"0""1"),可以先转换类型确保列名是数字:
df['ClasPri'] = df['ClasPri'].astype(int)
内容的提问来源于stack exchange,提问作者Vinicius Silva
相关产品推荐
相关产品推荐

