python - 如何返回不包括某些列的新数据框？

Question

我正在尝试获取数据框 df并返回一个新的数据框，不包括任何带有单词 'job' 的列在其名称中，排除任何带有字符串 'birth' 的列在其名称中，并排除这些列:name , userID , lgID .

我怎样才能做到这一点？

Answer 1

您可以使用 Pandas str 方法创建 bool 掩码，然后过滤 pd.Index 对象:

L = ['jobs', 'careers', 'birth', 'birthdate', 'name', 'userID', 'lgID', 'value1', 'value2']
df = pd.DataFrame(columns=L)

m1 = df.columns.str.contains('job|birth')         # regex supported
m2 = df.columns.isin(['name', 'userID', 'lgID'])  # use isin for exact matches

idx = df.columns[~(m1 | m2)]

Index(['careers', 'value1', 'value2'], dtype='object')

然后过滤您的数据框:

df = df[idx]