Как сделать произведение массивов из массива в Python

Question

Как в одной строке я могу получить произведение массивов массива? Мне нужно сделать это для случаев с несколькими столбцами * Пример 1001 *

2 столбцов:

X = [[1 4]
     [2 3]
     [0 2]
     [1 5]
     [3 1]
     [3 6]]

sol = [4 6 0 5 3 18]

Пример 4 столбцов:

X = [[1 4 2 3]
     [2 3 1 5]
     [0 2 3 4]
     [1 5 2 2]
     [3 1 1 6]
     [3 6 3 1]]

sol = [24 30 0 20 18 54]

Answer 1

Это умножение строк. Вы можете выполнить это с помощью:

X.prod(<b>axis=1</b>)

, например:

>>> X
array([[1, 4],
       [2, 3],
       [0, 2],
       [1, 5],
       [3, 1],
       [3, 6]])
>>> a.prod(axis=1)
array([ 4,  6,  0,  5,  3, 18])

Answer 2

Вы также можете использовать numpy .multiply.reduce

np.multiply.reduce(x, axis=1)