Ruby에서 두 배열의 조합

다음을 달성하는 Ruby 방법은 무엇입니까?

a = [1,2]
b = [3,4]

나는 배열을 원한다 :

=> [f(1,3) ,f(1,4) , f(2,3) ,f(2,4)]

ruby combinations

— Pierrotlefou
소스

@lucasarruda는 OP가 요구하는 것과 완전히 동일하지 않습니다. Zip은 [a[0], b[0]]및 의 배열을 생성합니다 [a[1], b[1]]. 포함되지 않습니다 [a[0], b[1]]. a.zip(b) => [[1,3],[2,4]]

— KNejad

감사합니다 @KNejad, 당신이 맞습니다. 내 댓글을 삭제했습니다.

— lucasarruda

답변:

147

를 사용 product하여 배열의 데카르트 곱을 먼저 얻은 다음 함수 결과를 수집 할 수 있습니다.

a.product(b) => [[1, 3], [1, 4], [2, 3], [2, 4]]

따라서 map또는 collect을 사용 하여 결과를 얻을 수 있습니다 . 동일한 방법에 대해 다른 이름입니다.

a.product(b).collect { |x, y| f(x, y) }

— 아론 힌니
소스

제품은 1.9부터 내장되어 있습니다.

— Pesto

제품은 1.8.7에 - 나는 Pierr 물어 정확히 어떻게 사용

— 크리스 맥컬리

@FloatingRock 잘못된 것은 permuation이 아닌 조합입니다. ruby-doc.org/core-2.1.3/Array.html#method-i-product

— Josh Diehl

@FloatingRock 그렇지 않습니다. 각 항목의 첫 번째 요소는 항상 첫 번째 목록에서 가져오고 두 번째 요소는 두 번째 목록에서 가져옵니다. 가능한 모든 요소 조합이지만 순서가 변경되지 않으므로 순열이 아닙니다.

— Mark Reed

그리고 만약 당신이 c = [5,6]와 d = [7,8]을 가지고 있다면 당신은 할 수 있습니다[a,b,c,d].reduce([[]]){|r,e| r=r.product(e)}.map{|x| x.flatten}

— xxjjnn

a.map {|x| b.map {|y| f(x,y) } }.flatten

참고 : 1.8.7+ 1에서는 flatten에 인수로 추가 할 수 있으므로 f배열을 반환 할 때 올바른 결과를 얻을 수 있습니다.

다음은 임의 개수의 배열에 대한 추상화입니다.

def combine_arrays(*arrays)
  if arrays.empty?
    yield
  else
    first, *rest = arrays
    first.map do |x|
      combine_arrays(*rest) {|*args| yield x, *args }
    end.flatten
      #.flatten(1)
  end
end

combine_arrays([1,2,3],[3,4,5],[6,7,8]) do |x,y,z| x+y+z end
# => [10, 11, 12, 11, 12, 13, 12, 13, 14, 11, 12, 13, 12, 13, 14, 13, 14, 15, 12, 13, 14, 13, 14, 15, 14, 15, 16]

— sepp2k
소스

나는 확실하지가 최선의 방법 인 경우 해요하지만이 작동합니다 (생각할 수 있지만 더 나은 하나 하나 정도)

— 로버트 Massaioli

임의의 양의 배열 조합에 대해 +2 수 있기를 바랍니다. 아주 멋지다!

— pierrotlefou

나도 이것으로 나왔다.하지만 그때 나는 그것을하기위한 더 쉬운 방법이 있어야한다는 것을 알고 있었다… 이것에 대해 +1 그러나 @Aaron Hinni는 그것의 클리너에 대답한다

— Orlando

패싯은 Array#product배열의 외적을 제공합니다. 또한 ** operator2-array 케이스의 별칭으로도 지정됩니다 . 그것을 사용하면 다음과 같이 보일 것입니다.

require 'facets/array'
a = [1,2]
b = [3,4]

(a.product b).collect {|x, y| f(x, y)}

Ruby 1.9를 사용 product하는 경우 내장 배열 함수입니다.

— 페스토
소스

왜 사람들은 Array # product가 1.9ism이라고 계속 말합니까? 적어도 1.8.7에 있습니다.

— Mark Reed