沿给定轴乘数组

时间:2018-07-11 13:51:29

标签: python numpy numpy-broadcasting

我目前正在按以下方式将两个具有不同维度的数组相乘

x = np.ones((100, 10, 10))
y = np.arange(0, 100)

for i in range(x.shape[1]):
    for j in range(x.shape[2]):
        x[:, i, j] *= y

我一直在考虑如何在不使用嵌套的for循环的情况下重写它,但是没有什么好主意。有没有办法使用numpy数组,或者至少没有嵌套循环,来实现此目的?

0 个答案:

没有答案