Pandas如何提取DataFrame指定列的行内容并生成嵌套列表

实现代码 #

import pandas as pd

# 构造匹配示例的DataFrame
d = {'a': [2, 3, 4], 'b': [3, 4, 5], 'c': [4, 5, 6]}
df = pd.DataFrame(data=d)

# 提取指定列后转嵌套列表
contents = df[['a', 'c']].to_numpy().tolist()
print(contents)
# 输出结果：[[2, 4], [3, 5], [4, 6]]

补充说明 #

• 低版本pandas可以把to_numpy()替换为values，写法为df[['a', 'c']].values.tolist()，效果一致
• 要调整提取的列，仅需要修改df[[]]内的列名列表即可，比如提取b、c列就改为df[['b', 'c']]

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