有没有一种有效的方法来生成所有可能的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之类的组合.