python - 如何在 DataFrame 中选择多个分散的列？

Question

我有一个包含 35 列的 DataFrame，我想创建一个包含其中 6 列的新 DataFrame。相关列分布在整个 DataFrame 的各个位置。最Pythonic 的方法是什么？

Answer 1

假设您有一个包含多列的数据框:

df = pd.DataFrame({'col1':[1,2,3], 'col2':[4,5,6],'col3':[7,8,9],
           'col4':[1,2,3],'col5':[1,2,3],'col6':[1,2,3],
           'col7':[1,2,3],'col8':[1,2,3],'col9':[1,2,3],
           'col91':[1,2,3]})
print(df)

输出:

   col1  col2  col3  col4  col5  col6  col7  col8  col9  col91
0     1     4     7     1     1     1     1     1     1      1
1     2     5     8     2     2     2     2     2     2      2
2     3     6     9     3     3     3     3     3     3      3

现在要使用选定的列创建新的dataframe，您只需将列名称列表传递给选定的['col1', 'col2', 'col5']:

tmp_df = df[['col1', 'col2', 'col5']]
print(tmp_df)

输出:

   col1  col2  col5
0     1     4     1
1     2     5     2
2     3     6     3

更多内容可以引用my other answer