python - 如何重新排列 Pandas 列序列？

Question

>>> df =DataFrame({'a':[1,2,3,4],'b':[2,4,6,8]})
>>> df['x']=df.a + df.b
>>> df['y']=df.a - df.b
>>> df
   a  b   x  y
0  1  2   3 -1
1  2  4   6 -2
2  3  6   9 -3
3  4  8  12 -4

现在我想重新排列列序列，使 'x','y' 列成为第一列和第二列:

>>> df = df[['x','y','a','b']]
>>> df
    x  y  a  b
0   3 -1  1  2
1   6 -2  2  4
2   9 -3  3  6
3  12 -4  4  8

但是，如果我有一个很长的 'a'、'b'、'c'、'd'......，并且我不想明确列出这些列。我该怎么做？

或者 Pandas 是否提供像 set_column_sequence(dataframe,col_name, seq) 这样的函数，以便我可以这样做:set_column_sequence(df,'x',0) 和 set_column_sequence(df,'y',1) ?

Answer 1

你也可以这样做:

df = df[['x', 'y', 'a', 'b']]

您可以通过以下方式获取列列表:

cols = list(df.columns.values)

输出将产生如下内容:

['a', 'b', 'x', 'y']

...然后很容易在将其放入第一个函数之前手动重新排列