如何从数据框中的选定列创建元组列表的变量列表？ [复制]答案

【问题标题】：How to create variable list of list of tuples from selected columns in dataframe? [duplicate]如何从数据框中的选定列创建元组列表的变量列表？ [复制]
【发布时间】：2019-05-19 03:56:58
【问题描述】：

我的 df 看起来像这样：

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 的数组并分配给变量？

示例最终结果：

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

【问题讨论】：

我在你的最终结果中没有看到任何元组
df[['col1', 'col2', 'col3']].values.tolist()
您可以选择特定列，然后转换为列表列表。查看标记的重复项。

标签： python arrays python-3.x pandas

【解决方案1】：

只需索引这些列，并使用.values 属性：

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

示例：

>>> 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]]

【讨论】：

成功了，谢谢！