如何将Pandas DataFrame转换为嵌套列表(List of List)?
没问题,这有几种简单高效的方法可以实现你的需求:
方法1:简洁高效的values.tolist()
这是最直接的方式,利用pandas内置方法一步完成转换:
import pandas as pd # 先构建你的DataFrame df = pd.DataFrame({'col1': ['a', 'b'], 'col2': [1, 2]}) # 转换为嵌套列表 nested_list = df.values.tolist() print(nested_list) # 输出结果:[['a', 1], ['b', 2]]
方法2:官方推荐的to_numpy().tolist()
如果使用的是较新版本的pandas,官方更推荐用to_numpy()替代values,语义更明确,效果完全一致:
nested_list = df.to_numpy().tolist()
方法3:逐行遍历(适合自定义处理场景)
如果需要对每一行做额外的自定义操作,可以用逐行遍历的方式,不过这种方法效率比前两种低,适合小数据集:
nested_list = [] for _, row in df.iterrows(): nested_list.append([row['col1'], row['col2']])
简单说明一下:前两种方法属于矢量化操作,处理大数据集时速度会快很多;第三种方法则更灵活,适合需要对每行数据做额外加工的场景。
内容的提问来源于stack exchange,提问作者kaulmonish
相关产品推荐
相关产品推荐

