按数字重新排序df.columns的字母

时间:2019-06-19 09:03:32

标签: python pandas sorting

我的df如下:

        Store   Spend_1 Spend_2 Spend_3 Spend_4 Variance_1  Variance_2  Variance_3  Variance_4
0   1   200 230 189 200 -14 16  -6  18
1   2   224 104 240 203 -17 -11 17  -18
2   3   220 168 131 210 10  -9  12  19
3   4   143 187 139 188 -1  -17 -20 -9
4   5   179 121 162 131 6   -25 5   20
5   6   208 158 140 191 16  -14 -22 -6

我试图在列名上应用自定义排序,以便按如下顺序进行排序:

    Store   Spend_1 Variance_1  Spend_2 Variance_2  Spend_3 Variance_3  Spend_4 Variance_4
0   1   200 -14 230 16  189 -6  200 18
1   2   224 -17 104 -11 240 17  203 -18
2   3   220 10  168 -9  131 12  210 19
3   4   143 -1  187 -17 139 -20 188 -9
4   5   179 6   121 -25 162 5   131 20
5   6   208 16  158 -14 140 -22 191 -6

我已经尝试过简单的sorted,但显然这适用于字母顺序,忽略了末尾的整数。

我以enumeratingnumbercols df.columns来变戏法,将字符串更改为整数,应用排序然后使用{{1 }},但我不确定如何以这种方式应用自定义排序。

有人可以帮助吗?

4 个答案:

答案 0 :(得分:3)

想法是通过两个值使用key参数,将_之后的值转换为inetegr前具有第一个值的_,但是解决方案适用于所有不带第一个值的列df.columns[1:],因此最后一个由df.columns[:1].tolist()的第一列添加:

cols = df.columns[:1].tolist() +sorted(df.columns[1:], 
                                       key=lambda x: (int(x.split('_')[1]), x.split('_')[0]))

df = df[cols]
print (df)
   Store  Spend_1  Variance_1  Spend_2  Variance_2  Spend_3  Variance_3  \
0      1      200         -14      230          16      189          -6   
1      2      224         -17      104         -11      240          17   
2      3      220          10      168          -9      131          12   
3      4      143          -1      187         -17      139         -20   
4      5      179           6      121         -25      162           5   
5      6      208          16      158         -14      140         -22   

   Spend_4  Variance_4  
0      200          18  
1      203         -18  
2      210          19  
3      188          -9  
4      131          20  
5      191          -6  

答案 1 :(得分:1)

您可以将key传递给sorted来进行自己的自定义排序:

sorted_columns = sorted(df.columns, key = lambda col: col[-1] + col[:-1])
df[sorted_columns]

想法是将最终整数放在第一位。如果您可以输入两位数的数字,则会崩溃。

答案 2 :(得分:1)

这是在_上拆分列,反转结果列表的一种方法,以便进一步排序优先于尾随数字并使用pandas.Index.argsort

df.iloc[:,[0]+[*df.columns.str.split('_').str[::-1].argsort()[:-1]]]

     Store  Spend_1  Variance_1  Spend_2  Variance_2  Spend_3  Variance_3  \
0      1      200         -14      230          16      189          -6   
1      2      224         -17      104         -11      240          17   
2      3      220          10      168          -9      131          12   
3      4      143          -1      187         -17      139         -20   
4      5      179           6      121         -25      162           5   
5      6      208          16      158         -14      140         -22   

     Spend_4  Variance_4  
0      200          18  
1      203         -18  
2      210          19  
3      188          -9  
4      131          20  
5      191          -6  

答案 3 :(得分:1)

我能想到的最简单的方法是定义自己的排序键

df = df.reindex(sorted(df.columns, key=lambda x: int(x.split("_")[1]) if "_" in x else 0), axis=1)