我的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
,但显然这适用于字母顺序,忽略了末尾的整数。
我以enumerating
为number
,cols
df.columns
来变戏法,将字符串更改为整数,应用排序然后使用{{1 }},但我不确定如何以这种方式应用自定义排序。
有人可以帮助吗?
答案 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)