LINUX.ORG.RU

Сложить несколько массивов всеми возможными комбинациями

 ,


0

1

Есть несколько неравных линейных массивов a, b, c, d…. Нужно посчитать суммы всех возможных комбинаций их элементов. Как их комбинировать? Очевидное решение, очевидно, и самое медленное:

sum_array = np.zeros( (len(a), len(b), ...) )
for i in range(len(a)):
    for j in range(len(b)):
.......
            sum_array[i, j, ...] = a[i] + b[j] + ...

Есть ли в numpy средства для такого суммирования?

★★★★★

Последнее исправление: question4 (всего исправлений: 1)

Ответ на: комментарий от vvn_black

Спасибо!

Вариант для 2: np.asarray(np.meshgrid(a, b)).T.sum(axis=2)

question4 ★★★★★
() автор топика
Последнее исправление: question4 (всего исправлений: 1)
Вы не можете добавлять комментарии в эту тему. Тема перемещена в архив.