使用itertools.combinations生成具有指定顺序的组合

Question

我使用itertools.combinations为数据帧的索引生成组合。我想要按指定顺序排列的组合-->(High - Mid - Low)

示例

from itertools import combinations
d = {'levels':['High', 'High', 'Mid', 'Low', 'Low', 'Low', 'Mid'], 'converted':[True, True, True, False, False, True, False]}
df = pd.DataFrame(data=d)
df_ = pd.crosstab(df['levels'], df['converted'])

df_

converted   False   True
levels      
High        0       2
Low         2       1
Mid         1       1

list(combinations(df_.index, 2))返回[('High', 'Low'), ('High', 'Mid'), ('Low', 'Mid')]

我希望第三组是('Mid', 'Low')，我如何实现这一点？

Answer 1

首先使用DataFrame.reindex，但列表中的第一个和第二个值被交换：

order = ['High','Mid','Low']
a = list(combinations(df_.reindex(order).index, 2))
print (a)
[('High', 'Mid'), ('High', 'Low'), ('Mid', 'Low')]