python - Numpy "double"-广播 - 有可能吗？

Question

是否可以使用“双”广播来删除以下代码中的循环？换句话说，要在整个时间数组 T 以及相同维度的数组 freqs 和 phases 上进行广播。

freqs = np.arange(100)
phases = np.random.randn(len(freqs))
T = np.arange(0, 500)

signal = np.zeros(len(T))
for i in xrange(len(signal)):
    signal[i] = np.sum(np.cos(freqs*T[i] + phases))

Answer 1

您可以通过添加新轴将 T reshape 为二维数组，这将在与一维数组相乘/相加时触发广播，然后再进行广播使用 numpy.sum 折叠此轴:

np.sum(np.cos(freqs * T[:,None] + phases), axis=1)
#                      add new axis        remove it with sum

<小时/>

测试:

(np.sum(np.cos(freqs * T[:,None] + phases), axis=1) == signal).all()
# True

​