按特定顺序对列名称进行排序

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

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

要获得其他所需的输出，只需在上面的代码中0与交换即可1。