Python基于每个lis的第一个值按自定义顺序排列的列表列表

Question

我想根据每个列表的第一个值按自定义顺序对列表进行排序。列表列表是pivot表dataframe（cols = list(dfOverview.columns.values)）的我的列，其中一些列本身就是一个列表。你知道吗

最简单的工作示例（我希望以b开头的列移到c之前）：

cols = ['2016', '2017', '2018', 
        ('a', '2016'), 
        ('a', '2017'), 
        ('a', '2018'), 
        ('c', '2016'), 
        ('c', '2017'), 
        ('c', '2018'), 
        ('b', '2016'), 
        ('b', '2017')]

sortOrder = {'2016': 0, '2017': 1, '2018': 2, 'a': 3, 'b': 4, 'c': 5}

cols.sort(key=lambda val: sortOrder[val[0]])

这返回了一个错误，即使我尝试了很多，google搜索了很多，我也找不到一个方法让它工作。cols的预期结果：

['2016', '2017', '2018', 
 ('a', '2016'), 
 ('a', '2017'), 
 ('a', '2018'),  
 ('b', '2016'), 
 ('b', '2017'),
 ('c', '2016'), 
 ('c', '2017'), 
 ('c', '2018')]

为了澄清我的问题的一些背景：我对一个pandas数据框进行了透视，得到了一个新的数据框，其中的列可以由一系列值组成（以多列作为参数的透视表）。这使得订购变得困难，因此出现了上述问题。你知道吗

我的数据帧是以下函数的结果：

dfPivot = df.pivot_table(index=['col1', 'col2'], 
                         columns=['year','category'], 
                         values='price')

Answer 1

这会有用的。它将sortkey设置为元组的第一个元素，否则设置为变量本身。你知道吗

cols = ['2016', '2017', '2018', 
        ('a', '2016'), 
        ('a', '2017'), 
        ('a', '2018'), 
        ('c', '2016'), 
        ('c', '2017'), 
        ('c', '2018'), 
        ('b', '2016'), 
        ('b', '2017')]

sorted(cols, key=lambda k: k[0] if isinstance(k, tuple) else k)

# ['2016',
#  '2017',
#  '2018',
#  ('a', '2016'),
#  ('a', '2017'),
#  ('a', '2018'),
#  ('b', '2016'),
#  ('b', '2017'),
#  ('c', '2016'),
#  ('c', '2017'),
#  ('c', '2018')]