python - 从列表值中获取所有组合

Question

我正在使用 python 2.7 并且我有这个列表:

new_out_filename = ['OFF_B8', 0, 'ON_B8', 1, 'ON_B16', 4, 'OFF_B0', 7]

我想获取字符串的所有组合，如 OFF_B8_vs_ON_B8、OFF_B8_vs_ON_B16、OFF_B8_vs_OFf_B0、ON_B8_vs_ON_16，等

有没有简单的方法可以实现？

我试过类似的东西:

for k in range(0, len(new_out_filename), 2):
    combination = new_out_filename[k]+'_vs_'+new_out_filename[k+2]
    print combination

但是我的列表超出了索引，而且我也没有得到合适的结果。

你能帮帮我吗？

Answer 1

只需在切片列表上使用组合即可忽略数字:

import itertools
new_out_filename = ['OFF_B8', 0, 'ON_B8', 1, 'ON_B16', 4, 'OFF_B0', 7]
for a,b in itertools.combinations(new_out_filename[::2],2):
    print("{}_vs_{}".format(a,b))

结果:

OFF_B8_vs_ON_B8
OFF_B8_vs_ON_B16
OFF_B8_vs_OFF_B0
ON_B8_vs_ON_B16
ON_B8_vs_OFF_B0
ON_B16_vs_OFF_B0

或理解:

result = ["{}_vs_{}".format(*c) for c in itertools.combinations(new_out_filename[::2],2)]

结果:

['OFF_B8_vs_ON_B8', 'OFF_B8_vs_ON_B16', 'OFF_B8_vs_OFF_B0', 'ON_B8_vs_ON_B16', 'ON_B8_vs_OFF_B0', 'ON_B16_vs_OFF_B0']