numpy.argwhere 与 numpy.where的区别

ar = np.array([1,2,3,4])
b = np.argwhere(ar==4)
print(type(b))
print(b)
#结果：
<class 'numpy.ndarray'>
[[3]]



ar = np.array([1,2,3,4])
b = np.where(ar==4)
print(type(b))
print(b)
#结果：
<class 'tuple'>
(array([3], dtype=int64),)