如何避免python中的“for”循环

Question

假设 X 是一个形状为 (3,8,8) 的 ndarray，而 y 是另一个形状为 (3,) 的数组。我想将 X 的每个 (8,8) 切片乘以 y 的一个元素，然后将结果相加。使用 for 循环，可以按如下方式完成：

import numpy as np
X = np.random.rand(3,8,8)
y = np.random.rand(3)
temp = 0
for k in range(3):
    temp = temp+y[k]*X[k,:,:]

有什么办法可以避免“for”循环吗？任何线索表示赞赏。

Answer 1

你有没有花足够的时间阅读 numpy 基础知识来遇到broadcasting？

(y[:,None,None]*X).sum(axis=0)

应该可以。

编辑

X 是 (3,8,8)。

y 需要是 (3,1,1) 才能使用它进行广播。

在尝试替换循环时，了解 broadcasting 至关重要。