Question

我已经尝试了下面的第二个例子并且它不起作用，但是一个小规模 - 将numpy数组与另一个数组分开的例子工作正常。有什么不同？我怎样才能让实际的部门工作？

我想要做的一个小规模的例子：

>>> import numpy as np
>>> a = np.asarray([[1, 2, 3], [4, 5, 6]])
>>> b = np.asarray([2, 2, 2])
>>> a / b           # works as expected
array([[0, 1, 1],
       [2, 2, 3]])

我正在努力工作的实际部门

>>> a = np.random.random((9, 31, 2, 5))
>>> b = np.random.random((9, 31, 2))
>>> a / b
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ValueError: operands could not be broadcast together with shapes (9,31,2,5) (9,31,2)

Answer 1

怎么样

a = np.random.random((9, 31, 2, 5))
b = np.random.random((9, 31, 2, 1))
print((a/b).shape)

产量

(9, 31, 2, 5)

划分不同形状的numpy多维数组

1 个答案: