来自前 2 个 Dataframe 列的 2 元素列表列表答案

【问题标题】：List of 2-element lists from first 2 Dataframe columns来自前 2 个 Dataframe 列的 2 元素列表列表
【发布时间】：2021-10-29 22:41:37
【问题描述】：

我有一个制表符分隔的.csv，现在我可以使用sep="\t" 打开它。

import pandas as pd

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

'test.csv' 看起来像这样：

value1 | value2 | value3
...

我想要一个 list 对象，即 2 元素长子列表的列表。这仅捕获前 2 列值。每个子列表代表数据框中的一行df：

my_list = [ [value1, value2], [value1, value2], [value1, value2] ] # ...

【问题讨论】：

【解决方案1】：

选择前两列并将输出转换为numpy array，然后转换为list：

L = df.iloc[:, :2].to_numpy().tolist()

【讨论】：