在熊猫中按标签选择多个列并非全部按顺序排列

Question

我想从数据帧中选择一组列，但这些列中只有一些是按顺序排列的。例如，我想写一些类似于：

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

这里也提出了类似的问题，但答案没有帮助：按标签选择多个列（熊猫）

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

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 ]]