如何将pandas pivot_table生成的透视表第一行转换为列表
pandas透视表取首行转列表实现方法
pandas的pivot_table生成透视表时,默认会对行索引按字母升序排序,你提供的示例中country字段排序后顺序为IT、UK、US,首行恰好是你需要的IT对应的数据,和预期输出匹配。
实现代码
方法1:用iloc按位置取首行(最直接)
你之前用iloc没成功大概率是忘记调用tolist()方法转列表,正确写法如下:
# 取透视表第0行(首行)转为列表 res = pivoted.iloc[0].tolist() print(res)
输出结果为:[9, 34, 15, 17, 18],完全符合预期。
方法2:按索引标签显式取值(更稳妥)
如果担心后续数据变化导致索引排序规则变动,也可以直接指定行索引标签取值,避免位置依赖:
# 显式取country为IT的行转为列表 res = pivoted.loc['IT'].tolist()
验证透视表结构
你可以执行以下代码查看透视表的完整结构,确认首行数据:
print(pivoted)
输出结构如下:
dimension a b c d e country IT 9 34 15 17 18 UK 10 9 34 29 10 US 29 10 9 34 29
内容的提问来源于stack exchange,提问作者dublin123
相关产品推荐
相关产品推荐

