Python如何在numpy数组中寻找特定的值?

1 个回答|60 次阅读
Coco老师 - 科普专家
擅长:编程教育

可以使用where()方法搜索数组

查找值为 4 的索引:

import numpy as np
arr = np.array([1, 2, 3, 4, 5, 4, 4])
x = np.where(arr == 4)
print(x)

执行结果:

(array([3, 5, 6], dtype=int64),)