从Dataframe中选择列

Question

我有一个包含许多列的数据框。我想将这个数据帧减少到只有我需要的列的数据框。

而不是使用

del df['column_name']

对于我不需要的所有列，有没有办法选择我做的那些并创建一个新的数据帧？我试过了：

df1['column_name'] = df['column_name']

Answer 1

您可以选择列表中的列：

cols = ['col1', 'col2']
df[cols]