numpy 查找 返回索引_numpy-查找操作大全

本文记录日常工作中遇到的查找操作，持续更新。

注意：输入必须是 数组，不能是 list

极值

min，max 返回极值

argmin(a, axis=None, out=None)， 返回极值所在的位置；不带 axis，先拉直，再找极值；带 axis，找某个维度的极值

b = np.array([[1, 2, 3, 5], [4, 6, 2, 6]])print(np.max(b)) #返回最大值 6

print(np.min(b)) #返回最小值 1

print(np.argmax(b)) #返回第一个最大值的位置 5

print(np.argmin(b)) #返回第一个最小值的位置 0

print(np.argmin(b, axis=1)) #[0 2]

NaN值

nan 值由多种表达形式，如 None，np.nan，np.NaN等

isnan，输入可以是 一维，也可以是 二维，返回布尔索引

x = np.array(range(10), dtype=np.float)

y= np.array(range(10,20))print(x.shape) #(10,)

print(x) #[0. 1. 2. 3. 4. 5. 6. 7. 8. 9.]

print(y) #[10 11 12 13 14 15 16 17 18 19]

x[3] = None #插入 nan

x[5] = np.NaN #插入 nan

print(x) #[ 0. 1. 2. nan 4. n