python - 当数据为 str 时，找出满足数值条件的索引

Question

我有这个数组

x = np.array([['1', 'Nb v'],
              ['2', '2'],
              ['3', 'Nb v'],
              ['4','3']])

我想找出第二列大于 2 的索引。这是第三个索引(最后一行)。

因为数据是字符串，所以我不能直接从字符串转换为 int，因为我在 Nb v 方面遇到问题。

所以，我想找出没有单词 Nb v 的索引。

idx, = np.where(x[:, 1] != 'Nb v')

给出:

array([1, 3])

然后找到值大于2的索引:

new_idx, = np.where(x[idx, 1].astype(int) > 2)

给出:

array([1])

但这给出了发生这种情况的 idx 索引，而不是 x 的索引。

我想找出 x 的索引，其中第二列是 > 2。

Answer 1

第二次不需要np.where，直接切片idx即可:

idx, = np.where(x[:, 1] != 'Nb v')
print (idx[x[idx, 1].astype(int) > 2])
# array([3], dtype=int64)