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

大佬教程收集整理的这篇文章主要介绍了使用 itertools.combinations 生成指定顺序的组合，大佬教程大佬觉得挺不错的，现在分享给大家，也给大家做个参考。

如何解决使用 itertools.combinations 生成指定顺序的组合？

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

示例

from itertools import combinations
d = {'levels':['High','High','MID','Low','MID'],'converted':[True,True,false,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')，我怎样才能做到这一点？

解决方法

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

大佬总结

以上是大佬教程为你收集整理的使用 itertools.combinations 生成指定顺序的组合全部内容，希望文章能够帮你解决使用 itertools.combinations 生成指定顺序的组合所遇到的程序开发问题。

如果觉得大佬教程网站内容还不错，欢迎将大佬教程推荐给程序员好友。

本图文内容来源于网友网络收集整理提供，作为学习参考使用，版权属于原作者。
如您有任何意见或建议可联系处理。小编QQ：384754419，请注明来意。