我有一個多維numpy陣列,我需要對其進行修改,以便將其元素修改為僅作為其中一個維度索引的函式。我當然可以用for回圈來做到這一點,如下面的簡化示例所示
import numpy as np
a = np.ones( (2,10) )
for ii in range(a.shape[1]):
a[:,ii] *= ii
如果陣列變得非常大,這可能會減慢執行速度,我想知道是否有一些巧妙的方法可以避免使用for回圈?
uj5u.com熱心網友回復:
構造另一個陣列來保存縮放因子,然后廣播和乘法:
scale = np.arange(a.shape[1])
a *= scale
轉載請註明出處,本文鏈接:https://www.uj5u.com/qita/361984.html
