Numpy「double」 - 廣播 - 有可能嗎？

Question

是否可以使用「double」 - 廣播去除下面的代碼中的循環？換句話說，要在整個時間陣列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作爲2D陣列/加入與一維數組，然後稍後使用numpy.sum摺疊此軸：

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

---

測試：

剛來找我（？但可能是計算昂貴）10

Answer 2

一個想法是構建參數爲矩陣：

phases = phases.reshape((len(phases), 1)) 
argumentMatrix = np.outer(freqs, T) + phases 
cosineMatrix = np.cos(argumentMatrix) 
signal = np.sum(cosineMatrix, axis=0) # sum, collapsing columns