python-2.7 - 从数据框中选择列

Question

我有一个包含许多列的数据框。我想将此数据框减少为一个仅包含我需要的列的数据框。

而不是使用

del df['column_name']

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

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

Answer 1

您可以选择列表中的列：

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