python - 根据列名中是否存在后缀对列进行排序

Question

我有一个像这样的数据框:

上面的df只是一个很小的样本。实际上我有大约 8K + 列，我想对我的数据框进行排序，以便所有以“_t1”结尾的列都在最后。

我绝对可以使用如下代码过滤掉数据帧的一个子集:

data = data [data.columns[data .columns.str.endswith("_t1")]]

然后合并。还有其他更简单的方法可以根据列名的正则表达式模式对 Pandas 数据框进行排序吗？

Answer 1

您可以创建面具:

m = data.columns.str.endswith("_t1")

或者用正则表达式屏蔽:

m = data.columns.str.contains("_t1$")

然后一起加入:

cols = data.columns[~m].append(data.columns[m])

或者:

cols = data.columns[~m].tolist() + data.columns[m].tolist()

并按子集更改列的顺序:

df = data[cols]