python - 在 Pandas 中按标签选择多列并非全部按顺序

Question

我想从 pandas 数据框中选择一组列，但其中只有一些列是按顺序排列的。例如，我想写这样的东西:

df.loc[:,['a','2001':'2040']]

此处提出了类似的问题，但答案没有帮助:Select multiple columns by labels (pandas)

就我而言，手动执行此操作太麻烦了。正则表达式是可能的但很复杂，因为切片是面向数字而不是面向文本的。那么最简单的方法是什么？

Answer 1

您可以使用列表理解:

In [11]: df.columns
Out[11]: Index(['a', '2001', '2002', '2040', '2041'], dtype='object')

In [12]: [c for c in df.columns if c == "a" or c.isdigit() and 2001 <= int(c) <= 2040 ]
Out[12]: ['a', '2001', '2002', '2040']

In [13]: df[[c for c in df.columns if c == "a" or c.isdigit() and 2001 <= int(c) <= 2040 ]]