performance - Numpy Index For ループを避ける

Question

forこのような操作に 2 番目のループを使用しないようにする方法はありますか?

for x in range(Size_1):
    for y in range(Size_2):
        k[x,y] = np.sqrt(x+y) - y

または、これを最適化するより良い方法はありますか? 現在、大きなサイズでは信じられないほど遅いです。

Answer 1

これはベクトル化されたソリューションですbroadcasting-

X,Y = np.ogrid[:Size_1,:Size_2]
k_out = np.sqrt(X+Y) - Y