关于python:如何从数据框中的选定列创建元组列表的变量列表?

How to create variable list of list of tuples from selected columns in dataframe?

本问题已经有最佳答案,请猛点这里访问。

我有这样的df:

1
2
3
4
5
6
col1     col2      col3     col4
1         0         1        0
1         0         1        0
1         0         1        0
1         0         1        0
1         0         1        0

如何将col1-col3转换为元组列表并赋值给变量?然后创建一个只有col4的数组并分配给变量?

最终结果示例:

1
2
lst = [[1,0,1],[1,0,1],[1,0,1],[1,0,1],[1,0,1]]]
lst_col4 = [[0],[0],[0],[0],[0]]


只需索引这些列,并使用.values属性:

1
lst = df[[col1, col2, col3]].values.tolist()

例子:

1
2
3
4
>>> df = pd.DataFrame({'a':[1,2,3], 'b':[4,5,6], 'c':[7,8,9]})
>>> lst = df[['a', 'c']].values.tolist()
>>> print(lst)
 [[1, 7], [2, 8],[3, 9]]