python - 选择/排除 Pandas 中的列集

Question

这个问题在这里已经有了答案:

Delete a column from a Pandas DataFrame (20 个回答)

关闭 3年前。

我想根据列选择从现有数据框创建 View 或数据框。

例如，我想从数据帧 df1 创建一个数据帧 df2，该数据帧包含其中的所有列，但其中两个除外。我尝试执行以下操作，但没有成功:

import numpy as np
import pandas as pd

# Create a dataframe with columns A,B,C and D
df = pd.DataFrame(np.random.randn(100, 4), columns=list('ABCD'))

# Try to create a second dataframe df2 from df with all columns except 'B' and D
my_cols = set(df.columns)
my_cols.remove('B').remove('D')

# This returns an error ("unhashable type: set")
df2 = df[my_cols]

我做错了什么？也许更一般地说，pandas 有什么机制来支持从数据帧中选择和排除任意列集？

Answer 1

您可以删除不需要的列或选择您需要的列

# Using DataFrame.drop
df.drop(df.columns[[1, 2]], axis=1, inplace=True)

# drop by Name
df1 = df1.drop(['B', 'C'], axis=1)

# Select the ones you want
df1 = df[['a','d']]