Как генерировать комбинации на python из двух групп чисел?

Question

[sha256]

Друзья, всем привет!
Пытаюсь сгенерировать набор данных по условию:
А = [1, 2]
B = [8, 9, 14]
Количество цифр в одной комбинации = 2

Мне необходимо получить уникальные комбинации без повторов где A выступает ключом, например:
1,8
1,9
1,14
или
2,8
2,9
2,14
таким образом не должно быть примеров без значений A.

Как это лучше сделать?

Пытаюсь сделать следующее:

import pandas as pd 
import numpy as np 
import itertools as itr, itertools

df = pd.DataFrame([[1,2], [7,8,14]])
ndf = df.apply(lambda x: list(itr.combinations(x,2)), axis = 1)
A = []
B = ndf.values.tolist()
for i in B:
  A.extend(i)

Но результат не соответствует ожиданиям:

[(1.0, 2.0), (1.0, nan), (2.0, nan), (7.0, 8.0), (7.0, 14.0), (8.0, 14.0)]

Comments

[Taus]

Приведите больше примеров комбинаций с количеством цифр в комбинации больше 2.

Answer 1

[Антон Федорян]

from itertools import product


A = [1, 2]
B = [8, 9, 14]
print(list(product(A, B)))  # [(1, 8), (1, 9), (1, 14), (2, 8), (2, 9), (2, 14)]

Comments

[sha256]

Спасибо, а как быть если нужно сгенерировать например:
1, 8, 9.
product - он ведь только пары генерит.

[Антон Федорян]

product - аналог вложенных циклов.

print(list(product([1], [8], [9]))) # [(1, 8, 9)]

[sha256]

Антон Федорян, понял, спасибо!