python – 生成零和b的所有可能组合

有没有一种有效的方法来生成所有可能的2个1和8个零组合的列表(或数组)?例如.

[[0,0,0,0,0,0,0,0,1,1],
 [0,0,0,0,0,0,0,1,0,1,],
 ...]

这有效,但可能有更好的方法吗?

import numpy as np
result = []
for subset in itertools.combinations(range(10), 2):
    subset = list(subset)
    c = np.zeros(10)
    c[subset] = 1
    result.append(c)

希望对如何优化此代码有一些想法.

解决方法:

嗯,它没有太大的不同,但在Numpy数组上进行批量操作肯定会有更少的开销:

import itertools
import numpy

which = numpy.array(list(itertools.combinations(range(10), 2)))
grid = numpy.zeros((len(which), 10), dtype="int8")

# Magic
grid[numpy.arange(len(which))[None].T, which] = 1

grid
#>>> array([[1, 1, 0, 0, 0, 0, 0, 0, 0, 0],
#>>>        [1, 0, 1, 0, 0, 0, 0, 0, 0, 0],
#>>>        [1, 0, 0, 1, 0, 0, 0, 0, 0, 0],
#>>>        [1, 0, 0, 0, 1, 0, 0, 0, 0, 0],
#>>>        [1, 0, 0, 0, 0, 1, 0, 0, 0, 0],
#>>> ...

大部分时间用于执行numpy.array(list(itertools.combinations(range(10),2))).我尝试使用numpy.fromiter,但我没有得到任何速度提升.由于一半的时间实际上是生成元组,因此进一步改进的唯一真正方法是生成C或Cython之类的组合.

上一篇:为什么python itertools“消耗”食谱比调用下n次更快?


下一篇:Python练习题答案: 真正的密码破解者【难度:2级】--景越Python编程实例训练营,1000道上机题等你来挑战