kết hợp giữa hai danh sách?

Đã được một thời gian và tôi gặp khó khăn trong việc xoay quanh một thuật toán mà tôi đang cố gắng thực hiện. Về cơ bản, tôi có hai danh sách và muốn nhận tất cả các kết hợp của hai danh sách.

Tôi có thể không giải thích nó chính xác vì vậy đây là một ví dụ.

name = 'a', 'b'
number = 1, 2

đầu ra trong trường hợp này sẽ là:

1.  A1 B2
2.  B1 A2

Phần khó khăn là tôi có thể có nhiều vật phẩm hơn trong biến số tên của tên lửa so với các vật phẩm trong biến số Số tên của con số (số sẽ luôn bằng hoặc ít hơn biến số tên).

Tôi bối rối làm thế nào để thực hiện tất cả các kết hợp (lồng nhau cho vòng lặp?) Và thậm chí còn bối rối hơn về logic để thay đổi các mục trong biến tên trong trường hợp có nhiều mục trong tên hơn trong danh sách số.

Tôi không phải là lập trình viên giỏi nhất nhưng tôi nghĩ tôi có thể thử nếu ai đó có thể giúp tôi làm rõ logic / algoriy nhịp để đạt được điều này. Vì vậy, tôi vừa bị mắc kẹt trên các vòng lặp lồng nhau.

Cập nhật:

Đây là đầu ra với 3 biến và 2 số:

name = 'a', 'b', 'c'
number = 1, 2

đầu ra:

1.  A1 B2
2.  B1 A2
3.  A1 C2
4.  C1 A2
5.  B1 C2
6.  C1 B2

Lưu ý : Câu trả lời này dành cho câu hỏi cụ thể được hỏi ở trên. Nếu bạn ở đây từ Google và chỉ tìm cách để có được một sản phẩm của Cartesian bằng Python, itertools.producthoặc một sự hiểu biết danh sách đơn giản có thể là những gì bạn đang tìm kiếm - hãy xem các câu trả lời khác.

Giả sử len(list1) >= len(list2). Sau đó, những gì bạn dường như muốn là lấy tất cả các hoán vị độ dài len(list2)từ list1và khớp chúng với các mục từ list2. Trong trăn:

import itertools
list1=['a','b','c']
list2=[1,2]

[list(zip(x,list2)) for x in itertools.permutations(list1,len(list2))]

Trả về

[[('a', 1), ('b', 2)], [('a', 1), ('c', 2)], [('b', 1), ('a', 2)], [('b', 1), ('c', 2)], [('c', 1), ('a', 2)], [('c', 1), ('b', 2)]]

Cách đơn giản nhất là sử dụng itertools.product:

a = ["foo", "melon"]
b = [True, False]
c = list(itertools.product(a, b))
>> [("foo", True), ("foo", False), ("melon", True), ("melon", False)]

Có thể đơn giản hơn cách đơn giản nhất ở trên:

>>> a = ["foo", "bar"]
>>> b = [1, 2, 3]
>>> [(x,y) for x in a for y in b]  # for a list
[('foo', 1), ('foo', 2), ('foo', 3), ('bar', 1), ('bar', 2), ('bar', 3)]
>>> ((x,y) for x in a for y in b)  # for a generator if you worry about memory or time complexity.
<generator object <genexpr> at 0x1048de850>

không có bất kỳ nhập khẩu

Tôi đã tìm kiếm một danh sách nhân với chỉ các kết hợp duy nhất, được cung cấp dưới dạng chức năng này.

import itertools
itertools.combinations(list, n_times)

Đây là một đoạn trích từ các tài liệu Python trên itertools Điều đó có thể giúp bạn tìm thấy những gì bạn đang tìm kiếm.

Combinatoric generators:

Iterator                                 | Results
-----------------------------------------+----------------------------------------
product(p, q, ... [repeat=1])            | cartesian product, equivalent to a 
                                         |   nested for-loop
-----------------------------------------+----------------------------------------
permutations(p[, r])                     | r-length tuples, all possible 
                                         |   orderings, no repeated elements
-----------------------------------------+----------------------------------------
combinations(p, r)                       | r-length tuples, in sorted order, no 
                                         |   repeated elements
-----------------------------------------+----------------------------------------
combinations_with_replacement(p, r)      | r-length tuples, in sorted order, 
                                         | with repeated elements
-----------------------------------------+----------------------------------------
product('ABCD', repeat=2)                | AA AB AC AD BA BB BC BD CA CB CC CD DA DB DC DD
permutations('ABCD', 2)                  | AB AC AD BA BC BD CA CB CD DA DB DC
combinations('ABCD', 2)                  | AB AC AD BC BD CD
combinations_with_replacement('ABCD', 2) | AA AB AC AD BB BC BD CC CD DD

Bạn có thể muốn thử hiểu một danh sách một dòng:

>>> [name+number for name in 'ab' for number in '12']
['a1', 'a2', 'b1', 'b2']
>>> [name+number for name in 'abc' for number in '12']
['a1', 'a2', 'b1', 'b2', 'c1', 'c2']

cách tốt nhất để tìm ra tất cả các kết hợp cho số lượng lớn danh sách là:

import itertools
from pprint import pprint

inputdata = [
    ['a', 'b', 'c'],
    ['d'],
    ['e', 'f'],
]
result = list(itertools.product(*inputdata))
pprint(result)

kết quả sẽ là:

[('a', 'd', 'e'),
 ('a', 'd', 'f'),
 ('b', 'd', 'e'),
 ('b', 'd', 'f'),
 ('c', 'd', 'e'),
 ('c', 'd', 'f')]

Hoặc câu trả lời KISS cho danh sách ngắn:

[(i, j) for i in list1 for j in list2]

Không phải là trình diễn như itertools nhưng bạn đang sử dụng python nên hiệu suất không phải là mối quan tâm hàng đầu của bạn ...

Tôi thích tất cả các câu trả lời khác quá!

một cải tiến nhỏ cho câu trả lời từ interjay, để đưa ra kết quả dưới dạng danh sách làm phẳng.

>>> list3 = [zip(x,list2) for x in itertools.permutations(list1,len(list2))]
>>> import itertools
>>> chain = itertools.chain(*list3)
>>> list4 = list(chain)
[('a', 1), ('b', 2), ('a', 1), ('c', 2), ('b', 1), ('a', 2), ('b', 1), ('c', 2), ('c', 1), ('a', 2), ('c', 1), ('b', 2)]

tham khảo từ liên kết này

Không có itertools

[(list1[i], list2[j]) for i in xrange(len(list1)) for j in xrange(len(list2))]

Trả lời câu hỏi "đưa ra hai danh sách, tìm tất cả các hoán vị có thể có của các cặp của một mục từ mỗi danh sách" và sử dụng chức năng Python cơ bản (nghĩa là không có itertools) và do đó, dễ dàng sao chép cho các ngôn ngữ lập trình khác:

def rec(a, b, ll, size):
    ret = []
    for i,e in enumerate(a):
        for j,f in enumerate(b):
            l = [e+f]
            new_l = rec(a[i+1:], b[:j]+b[j+1:], ll, size)
            if not new_l:
                ret.append(l)
            for k in new_l:
                l_k = l + k
                ret.append(l_k)
                if len(l_k) == size:
                    ll.append(l_k)
    return ret

a = ['a','b','c']
b = ['1','2']
ll = []
rec(a,b,ll, min(len(a),len(b)))
print(ll)

Trả về

[['a1', 'b2'], ['a1', 'c2'], ['a2', 'b1'], ['a2', 'c1'], ['b1', 'c2'], ['b2', 'c1']]

Các câu trả lời tốt hơn cho điều này chỉ hoạt động cho độ dài cụ thể của danh sách được cung cấp.

Đây là một phiên bản hoạt động cho bất kỳ độ dài đầu vào. Nó cũng làm cho thuật toán rõ ràng về các khái niệm toán học của sự kết hợp và hoán vị.

from itertools import combinations, permutations
list1 = ['1', '2']
list2 = ['A', 'B', 'C']

num_elements = min(len(list1), len(list2))
list1_combs = list(combinations(list1, num_elements))
list2_perms = list(permutations(list2, num_elements))
result = [
  tuple(zip(perm, comb))
  for comb in list1_combs
  for perm in list2_perms
]

for idx, ((l11, l12), (l21, l22)) in enumerate(result):
  print(f'{idx}: {l11}{l12} {l21}{l22}')

Kết quả này:

0: A1 B2
1: A1 C2
2: B1 A2
3: B1 C2
4: C1 A2
5: C1 B2