Python如何创建筛选器数组?

1 个回答|239 次阅读
Coco老师 - 公众号:量位学堂Qbit编程
擅长:编程教育

创建一个仅返回大于 10 的值的筛选器数组:

import numpy as np
arr = np.array([8, 10, 12, 21])
# 创建一个空列表
filter_arr = []
# 遍历arr
for element in arr:
  #如果元素大于10, 就将过滤数组设为True, 否则False:
  if element > 10:
    filter_arr.append(True)
  else:
    filter_arr.append(False)
newarr = arr[filter_arr]
print(filter_arr)
print(newarr)

执行结果:

[False, False, True, True]
[12 21]