如何用自己的坐标填充numpy数组

ykejflvf  于 2023-04-30  发布在  其他
关注(0)|答案(2)|浏览(125)

我有一个numpy nr x nc 整数数组,我想用它自己的数组坐标来填充每个元素,表示为2元组。例如:

(0,0)    (0,1)    ... (0,nc-1)
 ...      ...     ...    ...
(nr-1,0) (nr-1,1) ... (nr-1,nc-1)

我相信这可以通过列表理解来完成,但我不能让我的头周围如何,具体地。

lb3vh1jj

lb3vh1jj1#

列表解析的替代方法:您可以通过meshgrid创建一个坐标网格,并将其转换为网格:

i_coords, j_coords = np.meshgrid(range(nr), range(nc), indexing='ij')
coordinate_grid = np.array([i_coords, j_coords])

最后,你只需要将元素转换为元组。

vsmadaxz

vsmadaxz2#

我明白了

[ [(r,c) for c in range(nc) ] for r in range(nr) ]

相关问题