使用 np.random.normal 创建一个矩阵

Question

我需要创建一个 nxn 矩阵，其中单元格中的数字按照高斯分布分布。 此代码可能运行不顺利，因为它用序列填充单元格。 我该怎么办？

mu, sigma = 8, 0.5 # mean and standard deviation

def KHead(nx, ny, mu, sigma):
KH0=np.zeros((nx,ny))
N=1000
for k in range(1,ny-1):
    for i in range(0,nx-1):
        KH0[(i,k)]= np.random.normal(mu, sigma, N )

return KH0

Answer 1

为零边框编辑

np.random.normal 采用 size 关键字参数。 你可以这样使用它：

KH0 = np.zeros((nx, ny))
KH0[1:-1,1:-1] = np.random.normal(mu, sigma, (nx -2, ny - 2))