Question

我正在完成模拟Schelling的隔离模型的任务。

现在我想了解以下内容：

1）如何使用numpy生成100x100cells的二维矩阵（网格）

2）如何用40％的药剂A填充基质1; 40％的特工B与2;并且20％是空的，即0。代理是随机放置的。

我知道有numpy.ones：numpy.zeroes numpy.array但我不知道如何创建这样的2d矩阵并随机填充某些值。如何迭代？

有些人可能会帮助我吗？

谢谢！

Answer 1

有很多方法都有其优点和缺点。我不打算全部列出。

这个应该很容易理解，它不是最慢的，但可能也不是最快的，因为我们使用两个花哨的索引步骤（这里还有一个理由来实际阅读文档！）：

代码

import numpy as np

N = 10
a, b, c = 0.4, 0.4, 0.2

def build_mat(N, a, b, c):
    n_a = int(N * N * a)
    n_b = int(N * N * b)
    n_c = N - n_a - n_b

    rng_nonzero_inds_1d = np.random.choice(N*N, n_a + n_b, replace=False)

    mat = np.zeros(N*N)
    mat[rng_nonzero_inds_1d[:n_a]] = 1
    mat[rng_nonzero_inds_1d[n_a:]] = 2

    return mat.reshape((N,N))

mat = build_mat(N, a, b, c)
print('A: ', np.count_nonzero(mat==1))
print('B: ', np.count_nonzero(mat==2))
print('C: ', np.count_nonzero(mat==0))
print(mat)

输出

A:  40
B:  40
C:  20
[[ 1.  1.  2.  0.  2.  1.  0.  2.  1.  2.]
 [ 2.  1.  2.  1.  0.  1.  1.  0.  2.  0.]
 [ 0.  2.  1.  1.  2.  0.  2.  2.  2.  2.]
 [ 1.  0.  2.  1.  2.  2.  0.  1.  2.  0.]
 [ 1.  2.  1.  1.  2.  1.  2.  1.  2.  0.]
 [ 2.  0.  1.  2.  2.  1.  1.  2.  2.  1.]
 [ 0.  2.  1.  1.  2.  1.  1.  2.  0.  2.]
 [ 2.  2.  1.  2.  1.  2.  1.  1.  0.  0.]
 [ 2.  1.  1.  0.  0.  1.  2.  1.  1.  1.]
 [ 1.  1.  2.  2.  1.  2.  2.  0.  0.  1.]]

创建并初始化网格（矩阵）100x100 wih numpy

1 个答案:

代码

输出