python - 按月和年对数据框列进行排序

Question

我正在查看数据框中的每月数据。列标题都是字符串，从日期时间转换而来。这是我到目前为止的最终输出:

                             Feb 18  Jan 18  Mar 18
Flavor                                                      
Vanilla                      16.0    10.0     0.0
Chocolate                    20.0    20.0    16.0

似乎列是按字母顺序排序的。有没有一种方法可以按月和年对这些字符串类型的列进行排序？是否涉及使用前缀？

Answer 1

您可以将您的列名转为日期时间，然后对它们进行排序:

df.columns = pd.to_datetime(df.columns, format='%b %y')

df = df[sorted(df.columns)]

>>> df
           2018-01-01  2018-02-01  2018-03-01
Flavor                                       
Vanilla          10.0        16.0         0.0
Chocolate        20.0        20.0        16.0

如果您想要恢复原来的字符串列名称，只需添加:

df.columns = df.columns.strftime('%b %y')

>>> df
           Jan 18  Feb 18  Mar 18
Flavor                           
Vanilla      10.0    16.0     0.0
Chocolate    20.0    20.0    16.0

另一种方法是使用带键的 sorted 在一行中完成所有操作(感谢@SpghttCd 的想法):

df[sorted(df.columns, key = lambda x: pd.to_datetime(x, format='%b %y'))]