python - Pandas：如何获取除与给定列表匹配的列名之外的列名

Question

假设我有一个df：

df = pd.DataFrame({'day': range(1, 4),
                  'apple': range(5, 8),
                  'orange': range(7, 10),
                  'pear': range(9, 12),
                  'purchase': [1, 1, 1],
                  'cost': [50, 55, 60]})

day   apple   orange   pear   purchase   cost
1     5       7        9      1          50
2     6       8        10     1          55
3     7       9        11     1          60

如何获取所有列名但排除名称匹配的列名day, purchase, & cost？

标签： pythonpandas

Answer 1

通过列表理解

cols = [i for i in df.columns if i not in ["day", "purchase", "cost"]]