我有一个numpy nr x nc 整数数组,我想用它自己的数组坐标来填充每个元素,表示为2元组。例如:
(0,0) (0,1) ... (0,nc-1) ... ... ... ... (nr-1,0) (nr-1,1) ... (nr-1,nc-1)
我相信这可以通过列表理解来完成,但我不能让我的头周围如何,具体地。
lb3vh1jj1#
列表解析的替代方法:您可以通过meshgrid创建一个坐标网格,并将其转换为网格:
i_coords, j_coords = np.meshgrid(range(nr), range(nc), indexing='ij') coordinate_grid = np.array([i_coords, j_coords])
最后,你只需要将元素转换为元组。
vsmadaxz2#
我明白了
[ [(r,c) for c in range(nc) ] for r in range(nr) ]
2条答案
按热度按时间lb3vh1jj1#
列表解析的替代方法:您可以通过meshgrid创建一个坐标网格,并将其转换为网格:
最后,你只需要将元素转换为元组。
vsmadaxz2#
我明白了