Dataframe 按特定顺序对列名排序

假设pyspark数据帧有以下列名：

当然pyspark是按0、1、2等顺序排序的。但是，我想要的是：0_0；0_1; 1_0; 1_1; 2_0; 2_1 或者相反，0_0；1_0; 2_0; 3_0; 4_0; (...); 0_1; 1_1; 2_1; 3_1; 4_1 （我认为两种解决方案都可以）

---

有人能帮我吗？

您可以根据下划线前后的数字对列名进行排序：

df2 = df.select(
    'id',
    *sorted(
        df.columns[1:], key=lambda c: (int(c.split('_')[0]), int(c.split('_')[1]))
    )
)

---

要获得其他所需的输出，只需将上面代码中的

0

与

1

交换即可。

如何处理第一列（id）？因为它应该留在beggining@Johanna好的捕获，在编辑中修复