在 Python 中,我需要創建一個 NxM 矩陣,其中 ij 條目的值為 i^2 j^2。
我目前正在使用兩個for回圈構建它,但是陣列很大并且計算時間很長,我需要執行幾次。是否有使用 Numpy 構建此類矩陣的更有效方法?
uj5u.com熱心網友回復:
您可以在 numpy 中使用廣播。你可以參考官方檔案。例如,
import numpy as np
N = 3; M = 4 #whatever values you'd like
a = (np.arange(N)**2).reshape((-1,1)) #make it to column vector
b = np.arange(M)**2
print(a b) #broadcasting applied
np.arange()您可以使用np.array([...some array...])來自定義它,而不是使用。
轉載請註明出處,本文鏈接:https://www.uj5u.com/qukuanlian/533446.html
標籤:麻木的表现for循环矩阵
