numpy高级函数之where和extract函数

1   numpy.where() 函数返回输入数组中满足给定条件的元素的索引

----------------------------------------------------

代码：

n1=np.random.randint(10,20,10)

n2=np.where(n1>15)

结果：

[17 15 19 15 12 10 16 11 15 13]  #原始数组
(array([0, 2, 6], dtype=int64),)    #返回对象为一个元组，n2[0]是满足条件的索引，

-------------------------------

代码：

n1=np.random.randint(10,20,10)

n2=np.where(n1>15)

print(n1)

print(n2[0])

结果：

[15 17 12 10 11 12 17 11 12 19]
[1 6 9]#索引为1  6  9的元素满足条件

--------------------------------------------------------

代码：

n1=np.random.randint(10,20,10)

n2=np.where(n1>15,"big","small")

print(n1)

print(n2)

结果：

[10 18 16 10 17 14 12 11 16 18]
['small' 'big' 'big' 'small' 'big' 'small' 'small' 'small' 'big' 'big']

#逻辑判断操作：大于15的替换为"big"，否则替换为“small”

=====================================================

numpy.extract() 函数根据某个条件从数组中抽取元素，返回满条件的元素

代码：

n1=np.random.randint(10,20,10)

n3=np.extract(n1<15,n1)

print(n1)

print(n3)

结果：

[18 10 15 14 17 13 15 14 10 15]
[10 14 13 14 10]

#从n1中提取了小于15的数字