如何将含A、B列的Pandas DataFrame转为指定格式的列表?
将Pandas DataFrame转为元组列表
你可以通过以下几种简洁的方法实现需求:
方法1:使用itertuples()(高效且直观)
itertuples()会按行返回带索引的元组,只需剔除索引部分即可:
import pandas as pd data = [[0.8,0.23], [0.3,.40], [0.6,0.8], [0.7,0.9]] df = pd.DataFrame(data, columns=['A','B']) List = [tuple(row[1:]) for row in df.itertuples()] print(f"List= {List}")
输出:
List= [(0.8, 0.23), (0.3, 0.4), (0.6, 0.8), (0.7, 0.9)]
方法2:使用to_numpy()(简洁高效)
先将DataFrame转为numpy数组,再逐行转元组:
List = [tuple(x) for x in df.to_numpy()]
方法3:使用apply()(灵活易读)
按行应用tuple函数,再转为列表:
List = df.apply(tuple, axis=1).tolist()
以上三种方法都能得到你期望的格式,其中itertuples()和to_numpy()在处理大数据量时性能更优。
内容的提问来源于stack exchange,提问作者Pranab
相关产品推荐
相关产品推荐

