设置熊猫数据框中的列顺序


103

有没有一种方法可以根据我的个人喜好(即不按字母或数字排序,而是更像遵循某些约定)对熊猫数据框中的列进行重新排序?

简单的例子:

frame = pd.DataFrame({
        'one thing':[1,2,3,4],
        'second thing':[0.1,0.2,1,2],
        'other thing':['a','e','i','o']})

产生这个:

   one thing other thing  second thing
0          1           a           0.1
1          2           e           0.2
2          3           i           1.0
3          4           o           2.0

但是,我想这样:

   one thing second thing  other thing
0          1           0.1           a
1          2           0.2           e
2          3           1.0           i
3          4           2.0           o

(请提供通用解决方案,而不是针对此情况。非常感谢。)

Answers:


156

只需输入列名称即可自己选择订单。请注意双括号:

frame = frame[['column I want first', 'column I want second'...etc.]]

24
这仅适用于这个相当小的示例。如果要从其他来源(例如csv文件或数据库表)读取数据,则无法使用此答案。而且这些似乎更为普遍。OP要求一个总体解决方案。
基督

83

您可以使用此:

columnsTitles = ['onething', 'secondthing', 'otherthing']

frame = frame.reindex(columns=columnsTitles)

5
即使大多数其他解决方案都更为简洁,但对于那些不是100%熟悉的人来说,我认为它是最具可读性的pandas
德克

3
请记住,尽管要将返回值分配给变量,但这不会就位修改列顺序(至少在pandasv0.23` 中不会)。
德克

感谢@Dirk的建议
Okroshiashvili '18

33

这是我经常使用的解决方案。当您拥有包含大量列的大型数据集时,您绝对不希望手动重新排列所有列。

您可以而且很可能想做的是只是对您经常使用的前几列进行排序,而让所有其他列成为自己。这是R中的常用方法。df %>%select(one, two, three, everything())

因此,您可以首先手动键入要排序的列和要位于列表中所有其他列之前的列cols_to_order

然后,通过组合其余各列来构造新列的列表:

new_columns = cols_to_order + (frame.columns.drop(cols_to_order).tolist())

之后,您可以使用new_columns建议的其他解决方案。

import pandas as pd
frame = pd.DataFrame({
    'one thing': [1, 2, 3, 4],
    'other thing': ['a', 'e', 'i', 'o'],
    'more things': ['a', 'e', 'i', 'o'],
    'second thing': [0.1, 0.2, 1, 2],
})

cols_to_order = ['one thing', 'second thing']
new_columns = cols_to_order + (frame.columns.drop(cols_to_order).tolist())
frame = frame[new_columns]

   one thing  second thing other thing more things
0          1           0.1           a           a
1          2           0.2           e           e
2          3           1.0           i           i
3          4           2.0           o           o

1
辉煌,完美。感谢您让我不必键入每个列名或索引
stuart

这是一般性的答案,应该是公认的答案
CarlosH

26

您也可以做类似的事情 df = df[['x', 'y', 'a', 'b']]

import pandas as pd
frame = pd.DataFrame({'one thing':[1,2,3,4],'second thing':[0.1,0.2,1,2],'other thing':['a','e','i','o']})
frame = frame[['second thing', 'other thing', 'one thing']]
print frame
   second thing other thing  one thing
0           0.1           a          1
1           0.2           e          2
2           1.0           i          3
3           2.0           o          4

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

cols = list(df.columns.values)

输出将产生如下内容:

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

这样就很容易手动重新排列。


13

用列表而不是字典构造它

frame = pd.DataFrame([
        [1, .1, 'a'],
        [2, .2, 'e'],
        [3,  1, 'i'],
        [4,  4, 'o']
    ], columns=['one thing', 'second thing', 'other thing'])

frame

   one thing  second thing other thing
0          1           0.1           a
1          2           0.2           e
2          3           1.0           i
3          4           4.0           o

我无法获得“列名”:数据在列表中像字典中一样工作。
金·米勒

10

您还可以使用OrderedDict:

In [183]: from collections import OrderedDict

In [184]: data = OrderedDict()

In [185]: data['one thing'] = [1,2,3,4]

In [186]: data['second thing'] = [0.1,0.2,1,2]

In [187]: data['other thing'] = ['a','e','i','o']

In [188]: frame = pd.DataFrame(data)

In [189]: frame
Out[189]:
   one thing  second thing other thing
0          1           0.1           a
1          2           0.2           e
2          3           1.0           i
3          4           2.0           o

6

添加“ columns”参数:

frame = pd.DataFrame({
        'one thing':[1,2,3,4],
        'second thing':[0.1,0.2,1,2],
        'other thing':['a','e','i','o']},
        columns=['one thing', 'second thing', 'other thing']
)

4

尝试建立索引(因此您不仅需要通用的解决方案,因此索引顺序也可以是您想要的):

l=[0,2,1] # index order
frame=frame[[frame.columns[i] for i in l]]

现在:

print(frame)

是:

   one thing second thing  other thing
0          1           0.1           a
1          2           0.2           e
2          3           1.0           i
3          4           2.0           o

-2

我发现这是最简单,最有效的方法:

df = pd.DataFrame({
        'one thing':[1,2,3,4],
        'second thing':[0.1,0.2,1,2],
        'other thing':['a','e','i','o']})

df = df[['one thing','second thing', 'other thing']]
By using our site, you acknowledge that you have read and understand our Cookie Policy and Privacy Policy.
Licensed under cc by-sa 3.0 with attribution required.