使用Pandas读取制表符分隔CSV后提取前两列生成二维子列表的方法

实现方法 #

你可以通过pandas的位置索引选取前两列，再调用tolist()方法直接得到目标格式的列表，完整代码如下：

import pandas as pd

col_names = ['instance', 'class', 'example']
df = pd.read_csv('test.csv', sep="\t", names=col_names)

# 核心转换代码
my_list = df.iloc[:, :2].to_numpy().tolist()

代码说明 #

• df.iloc[:, :2]：选取DataFrame的所有行，以及前两列（列索引为0、1的instance和class列）
• to_numpy()：将选中的DataFrame片段转换为numpy数组
• tolist()：将numpy数组转换为嵌套列表，输出结构完全符合你要求的[[value1, value2], [value1, value2], ...]格式

如果你使用的是较老版本的pandas，也可以用df.iloc[:, :2].values.tolist()实现相同效果。

内容的提问来源于stack exchange，提问作者user16341274