0과 b의 모든 가능한 조합 생성

Question

말 2와 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) 

이 코드를 최적화하는 방법에 대한 아이디어가 있습니다.

Answer 1

글쎄, 그것은 훨씬 적은 오버 헤드를 가지고 바인딩 많이 다르지만 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과 같은 것으로 조합을 생성하는 것입니다.

Answer 2

numpy.bincount 사용하여 대안 :

>>> [np.bincount(xs, minlength=10) for xs in itertools.combinations(range(10), 2)] 
[array([1, 1, 0, 0, 0, 0, 0, 0, 0, 0], dtype=int64), 
array([1, 0, 1, 0, 0, 0, 0, 0, 0, 0], dtype=int64), 
array([1, 0, 0, 1, 0, 0, 0, 0, 0, 0], dtype=int64), 
array([1, 0, 0, 0, 1, 0, 0, 0, 0, 0], dtype=int64), 
...] 

Answer 3

것은 우리가 이것에 대한 순열을 사용하지할까요? 예 :

from itertools import permutations as perm 
a, b = 6, 2 
print '\n'.join(sorted([''.join(s) for s in set(t for t in perm(a*'0' + b*'1'))]))