根據Bool值挑選陣列中元素

覺得有用的話,歡迎一起討論相互學習~Follow Me

使用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_select\n",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]