python - 根据 Pandas 数据帧中的序列重新排列列

我有一个熊猫数据框，如下所示。我想根据XX_和YY_列的顺序分别重新排列数据框中的列。

import numpy as np
import pandas as pd
import math
import sys
import re
data=[[np.nan,2, 5,np.nan,np.nan,1],
      [np.nan,np.nan,2,np.nan,np.nan,np.nan],
      [np.nan,3,np.nan,np.nan,np.nan,np.nan],
      [1,np.nan,np.nan,np.nan,np.nan,1],
      [np.nan,2,np.nan,np.nan,2,np.nan],
      [np.nan,np.nan,np.nan,2,np.nan,5]]
df = pd.DataFrame(data,columns=['XX_4','XX_2','XX_3','YY_4','YY_2','YY_3'])
df

我的输出数据框应如下所示：

   XX_2  XX_3  XX_4  YY_2  YY_3  YY_4
0   2.0   5.0   NaN   NaN   1.0   NaN
1   NaN   2.0   NaN   NaN   NaN   NaN
2   3.0   NaN   NaN   NaN   NaN   NaN
3   NaN   NaN   1.0   NaN   1.0   NaN
4   2.0   NaN   NaN   2.0   NaN   NaN
5   NaN   NaN   2.0   NaN   5.0   2.0

由于这是一个很小的数据框，因此我可以手动重新排列列。有什么办法基于_2，_3后缀吗？

最佳答案

IIUC我们可以使用基于Jeff Attwood's文章的功能来对Mark Byers编写的字母数字列进行排序：

https://stackoverflow.com/a/2669120/9375102

import re
def sorted_nicely( l ):
    """ Sort the given iterable in the way that humans expect."""
    convert = lambda text: int(text) if text.isdigit() else text
    alphanum_key = lambda key: [ convert(c) for c in re.split('([0-9]+)', key) ]
    return sorted(l, key = alphanum_key)

df = pd.DataFrame(data,columns=['XX_9','XX_10','XX_3','YY_9','YY_10','YY_3'])
data = df.colums.tolist()
print(df[sorted_nicely(data)])
 XX_3  XX_9  XX_10  YY_3  YY_9  YY_10
0   5.0   NaN    2.0   1.0   NaN    NaN
1   2.0   NaN    NaN   NaN   NaN    NaN
2   NaN   NaN    3.0   NaN   NaN    NaN
3   NaN   1.0    NaN   1.0   NaN    NaN
4   NaN   NaN    2.0   NaN   NaN    2.0
5   NaN   NaN    NaN   5.0   2.0    NaN