python - 将 pandas df 列转换为配对列表

Question

这个问题已经有答案了:

Pandas DataFrame to List of Lists (14 个回答)

已关闭 4 年前。

如果我有这样的 df:

df1 = pd.DataFrame({'col1': [1, 1, 2, 2],
                   'col2': [10, 20, 10, 20]})

如何获得一个将 col1 的每一行配对的列表和col2像这样:

outputlist = [[1, 10], [1, 20], [2, 10], [2, 20]]

我找到了将列表转换为 df 列的方法，但反之则不然!

Answer 1

你可以这样做(参见tolist):

import pandas as pd

df1 = pd.DataFrame({'col1': [1, 1, 2, 2],
                   'col2': [10, 20, 10, 20]})

result = df1.values.tolist()
print(result)

输出

[[1, 10], [1, 20], [2, 10], [2, 20]]