import numpy as np
a = np.array([1,2,3,4,5,6,7,8,9,10])
b = np.array([1,0,1,0,1,0,1,0,1,0])
a = a[ b == 1]
print(a)
# output:[1 3 5 7 9]
numpy 按0,1数组取元素
最新推荐文章于 2022-10-03 22:40:36 发布
import numpy as np
a = np.array([1,2,3,4,5,6,7,8,9,10])
b = np.array([1,0,1,0,1,0,1,0,1,0])
a = a[ b == 1]
print(a)
# output:[1 3 5 7 9]