Numpy根据Bool值挑选数组中元素

时间:2022-07-23
本文章向大家介绍Numpy根据Bool值挑选数组中元素，主要内容包括其使用实例、应用技巧、基本知识点总结和需要注意事项，具有一定的参考价值，需要的朋友可以参考一下。

使用Boolean类型的数组挑选一维数组中的值

  • 使用一维Boolean数组选取数组中的特定元素，对应位置为True则选取，为False则不选取
import numpy as np
i_=[2]  # 挑选第三行数据
pop=np.arange(12).reshape(3,4)
print("pop(3,4)n",pop)
bool=np.array([0,0,1,1]).astype("bool") # [False,False,True,True]
# 使用数组作为索引选择个体
pop_i_=pop[i_]
print("pop_i_n",pop_i_)
#选择个体中选定的位置
pop_select=pop[i_,bool]
print("pop_selectn",pop_select)

# pop(3,4)
#  [[ 0  1  2  3]
#  [ 4  5  6  7]
#  [ 8  9 10 11]]
# pop_i_
#  [[ 8  9 10 11]]
# pop_select
#  [10 11]

随机文章
本站知识点必读
最近更新