《NumPy快速入门》11 - 网格

188 阅读1分钟

用于初始化

image.png

[[(j - i) for j in range(3)] for i in range(2)]

image.png

i, j = np.arange(2), np.arange(3)
I, J = np.meshgrid(i, j, indexing='ij');
J - I

image.png

i, j = np.arange(2), np.arange(3)
I, J = np.meshgrid(i, j, sparse=True, indexing='ij');
J - I

用于索引

image.png

a = np.array([[1,2], [3,4]])
I, J = np.indices(a.shape)
a[J,I]