從數據框中選擇列

Question

我有一個包含許多列的數據框。我想將這個數據框減少到只有我需要的列的數據框。

而不是使用

del df['column_name'] 

的，我並不需要的所有列的，是有辦法來選擇那些我做的，創建一個新的數據幀？我曾嘗試：

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

Answer 1

您可以在列表中選擇列：

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