array-布尔类型索引

问题：给一个数组，选出数组中所有大于5的数

import numpy as np
import random

a = [random.randint(0,10) for i in range(20)]
print("a的结果是:",a)

#第一种方式
b = list(filter(lambda x:x > 5,a))
print("b的结果是:",b)

#第二种方式
a = np.array(a)
c = a[a > 5]
print(c,type(c))

image.png

原理

image.png

求取大于5的偶数

d = a[(a>5) & (a%2 == 0)]
print("大于5的偶数为:",d)

image.png