Zip неровные Numpy массивы

Question

Рассмотрим следующие numpy.arrays:

a = np.array([1., 2., 3.]) 
b = np.array([4., 5.]) 
c = np.array([6., 7.]) 

мне нужно совместить эти, так что я в конечном итоге со следующим:

[(1., 4., 6.), (1., 5., 7.), (2., 4., 6.), (2., 5., 7.), (3., 4., 6.), (3., 5., 7.)] 

Обратите внимание, что в этом случае массив a случается быть самым большим массивом. Однако это не гарантировано. Также не гарантируется длина. Другими словами, любой массив может быть самым длинным, и каждый массив имеет произвольную длину.

Я пробовал использовать itertools.izip_longest, но я могу использовать только fillvalue для кортежа с 3., который не сработает. Я также попробовал itertools.product, но мой результат не является истинным декартовым продуктом.

Answer 1

Вы можете транспонировать b и c, а затем создать продукт a с транспонированной матрицы с помощью itertools.product:

>>> from itertools import product 
>>> [np.insert(j,0,i) for i,j in product(a,np.array((b,c)).T)] 
[array([ 1., 4., 6.]), array([ 1., 5., 7.]), array([ 2., 4., 6.]), array([ 2., 5., 7.]), array([ 3., 4., 6.]), array([ 3., 5., 7.])] 
>>> 

Answer 2

Допустим, у вас есть:

a = np.array([4., 5.]) 
b = np.array([1., 2., 3.]) 
c = np.array([6., 7.]) 
d = np.array([5., 1]) 
e = np.array([3., 2.]) 

Теперь, если вы знаете, прежде чем , который является самым длинным массивом, который равен b, в этом случае вы можете использовать подход, основанный на np.meshgrid -

# Concatenate elements from identical positions from the equal arrays 
others = np.vstack((a,c,d,e)).T # If you have more arrays, edit this line 

# Get grided version of the longest array and 
# grided-indices for indexing into others array 
X,Y = np.meshgrid(np.arange(others.shape[0]),b) 

# Concatenate grided longest array and grided indexed others for final output 
out = np.hstack((Y.ravel()[:,None],others[X.ravel()])) 

Пример запуска -

In [47]: b 
Out[47]: array([ 1., 2., 3.]) 

In [48]: a 
Out[48]: array([ 4., 5.]) 

In [49]: c 
Out[49]: array([ 6., 7.]) 

In [50]: d 
Out[50]: array([ 5., 1.]) 

In [51]: e 
Out[51]: array([ 3., 2.]) 

In [52]: out 
Out[52]: 
array([[ 1., 4., 6., 5., 3.], 
     [ 1., 5., 7., 1., 2.], 
     [ 2., 4., 6., 5., 3.], 
     [ 2., 5., 7., 1., 2.], 
     [ 3., 4., 6., 5., 3.], 
     [ 3., 5., 7., 1., 2.]]) 

Answer 3

Если различия в длине не являются экстремальными (проверьте входы первого) я был бы соблазн раздуть из коротких списков длины самой длинной с None и генерировать все перестановки (27 из них для 3 списков из 3 элементов). Тогда

results = [] 
for candidate in possibles: 
    if not (None in candidate): results.append(candidate) 

причин не делать этого: если куб длины самого длинного списка является существенным с точки зрения использования памяти (пространство для хранения N кубиками-плюсом) или использование центрального процессора.