我有与 A 和 B 相关联的数组,它们存储每个网格点的 (x,y) 位置。
在那种情况下,答案应该相当简单......
这两个网格是否严格遵循相同的网格方案?假设它们是,您可以执行以下操作:
np.argwhere((Ax == Bx.min()) & (Ay == By.min()))
假设两个网格的世界坐标与网格的索引方向相同,这给出了子集网格的左下角。(如果它们没有在同一方向上增加(即负dx
或dy
),它只会给出其他角之一)
在下面的示例中,我们显然可以从ix = (Bxmin - Axmin) / dx
等计算适当的索引,但假设您有一个更复杂的网格系统,这仍然有效。但是,这是假设两个网格在同一个网格方案上!如果它们不是,它会稍微复杂一些......
import numpy as np
# Generate grids of coordinates from a min, max, and spacing
dx, dy = 0.5, 0.5
# For the larger grid...
Axmin, Axmax = -180, 180
Aymin, Aymax = -90, 90
# For the smaller grid...
Bxmin, Bxmax = -5, 10
Bymin, Bymax = 30, 40
# Generate the indicies on a 2D grid
Ax = np.arange(Axmin, Axmax+dx, dx)
Ay = np.arange(Aymin, Aymax+dy, dy)
Ax, Ay = np.meshgrid(Ax, Ay)
Bx = np.arange(Bxmin, Bxmax+dx, dx)
By = np.arange(Bymin, Bymax+dy, dy)
Bx, By = np.meshgrid(Bx, By)
# Find the corner of where the two grids overlap...
ix, iy = np.argwhere((Ax == Bxmin) & (Ay == Bymin))[0]
# Assert that the coordinates are identical.
assert np.all(Ax[ix:ix+Bx.shape[0], iy:iy+Bx.shape[1]] == Bx)
assert np.all(Ay[ix:ix+Bx.shape[0], iy:iy+Bx.shape[1]] == By)