在2个二维数组上使用Numpy Where

Question

我正在尝试这段代码，但在Numpy Where部分遇到了问题：

import numpy as N
....
....
sfolat = N.ravel(N.where((lat>37.5689) & (lat<37.6689)))
sfolon = N.ravel(N.where((lon>-122.4250) & (lon<-122.3250)))
sfocoord = N.ravel(N.where((lat>37.5189) & (lat<37.7189)&(lon>-122.4750) & (lon<-122.2750)))

sfocoord返回

>>>sfocoord
array([204, 204, 205, 205, 145, 146, 145, 146])

lat和lon都有尺寸（428614）。我正试图在37.6189，-122.3750左右找到位置我想输入要更改的内容，这样代码就可以工作了。

Answer 1

numpy.where将返回一个长度为2的元组，其中2个元素为：行的索引数组和满足条件的列的相应索引数组。

numpy.ravel将把两个数组的元组扁平化为一个数组，并且行和列索引将不再有两个不同的数组。

保存索引：

idx = numpy.where((lat>37.5689) & (lat<37.6689)&(lon>-122.4250) & (lon<-122.3250))

根据sfocoord的给定输出，您的代码可能会输出

>>>idx 
(array([204, 204, 205, 205]), array([145, 146, 145, 146]))

CCD_ 4是行索引，CCD_。

要使用这些索引从lat数组中获取值，可以执行以下操作：

lat[idx[0], idx[1]]

编辑：一种将索引视为行、列对的方法：

>>>numpy.transpose(idx)
array([[204, 145],
   [204, 146],
   [205, 145],
   [205, 146]])