배열 집합의 요소 조합을 모두 찾는 방법은 무엇입니까?

Question

다음과 같은 복잡한 문제를 단순화하고 있습니다 ...

세 개의 정수 배열을 가정 할 때 각 요소의 가능한 모든 조합을 반환하는 가장 효율적인 방법은 무엇입니까? 각 배열의 각 값은 항상 동일한 위치에 지정되므로 [A,B,C]은 [C,B,A]과 같을 것입니다. 원하는 결과는 각 해시가 단일 조합을 포함하는 배열의 배열입니다. 예를 들어 :

을 감안할 때 :

는

var array1 = [1,2] 
var array2 = [a,b] 
var array3 = [foo,bar] 

이 결과는 다음과 같습니다

itertools.product(array1, array2, array3) 

을하고 필요한 경우 list에 포장 :

이

[ 
    [1,a,foo], 
    [2,a,foo], 
    [1,b,foo], 
    [2,b,foo], 
    [1,a,bar], 
    [2,a,bar], 
    [1,b,bar], 
    [2,b,bar] 
] 

Answer 1

파이썬에서, itertools.product를 사용 반복 가능하지 않은 시퀀스. 코드의 버전은 특히 효율적이지 않지만

def product(*args, **kwds): 
    # product('ABCD', 'xy') --> Ax Ay Bx By Cx Cy Dx Dy 
    # product(range(2), repeat=3) --> 000 001 010 011 100 101 110 111 
    pools = map(tuple, args) * kwds.get('repeat', 1) 
    result = [[]] 
    for pool in pools: 
     result = [x+[y] for x in result for y in pool] 
    for prod in result: 
     yield tuple(prod) 

:

당신이 어떻게하는지보고 싶다면

이는 itertools 문서에 나와있는 "해당"코드입니다.

PyPy version of itertools에는 파이썬 구현도 있습니다.

Answer 2

여기 파이썬에서 그 일의 다소 자세한 방법 :

# define the sets of items 
array1 = [1,2] 
array2 = ['a', 'b'] 
array3 = ['foo', 'bar'] 

# create an empty list to collect combinations 
all_combinations = [] 

# enumerate all combinations by nested iteration 
for i in array1: 
    for j in array2: 
     for k in array3: 
      all_combinations.append([i, j, k]) 

# print all combinations 
for item in all_combinations: 
    print item 

Answer 3

itertools.product 직접이 문제를 해결하고 매우 빠른 :

: 여기

>>> from itertools import product 
>>> list(product([1,2], ['a', 'b'], ['foo', 'bar'])) 
[(1, 'a', 'foo'), (1, 'a', 'bar'), (1, 'b', 'foo'), (1, 'b', 'bar'), (2, 'a', 'foo'), (2, 'a', 'bar'), (2, 'b', 'foo'), (2, 'b', 'bar')] 

Answer 4

자바 스크립트의 솔루션입니다

function p(o){ 
    var count=1; 
    var step_len=[]; 
    for(var i=0;i<o.length;i++){ 
     step_len[i]=count; 
     count*=o[i].length; 
    } 
    for(var i=0;i<count;i++){ 
     var tmp=[]; 
     for(var n=0;n<o.length;n++){ 
      tmp.push(o[n][Math.floor(i/step_len[n])%o[n].length]); 
     } 
     console.log(tmp); 
    } 
} 

var o=[ 
    [1,2], 
    ['a','b'], 
    ['foo','bar'] 
]; 

p(o); 

/* console output: 
[1, "a", "foo"] 
[2, "a", "foo"] 
[1, "b", "foo"] 
[2, "b", "foo"] 
[1, "a", "bar"] 
[2, "a", "bar"] 
[1, "b", "bar"] 
[2, "b", "bar"] 
*/ 

모든 수의 어레이에서 작동합니다.

데모 :

Answer 5

http://jsfiddle.net/u9zJQ/는 @의 stewe의 대답의 파이썬 equivalant했다. 나는이 대답이 더 좋고 itertools를 사용하는 어떤 대답이라도 더 일반적이고 모든 언어에 적용 할 수 있습니다. VBA에서이 기능이 필요했고 매력처럼 작동했습니다. +1

def p(o): 
    count=1 
    step_len=[] 
    for i, item in enumerate(o): 
     step_len.append(count) 
     count *= len(o[i]) 

    for i in range(count): 
     tmp=[] 
     for n, item in enumerate(o): 
      tmp.append(o[n][(i//step_len[n]) % len(o[n])]) 

     print tmp 
    print count