------布尔索引-----

布尔索引是指在 numpy 数组中使用布尔值(TrueFalse)作为索引来选择数组中的元素。

在使用布尔索引时,首先需要创建一个布尔型数组,其中数组的每个元素都是 TrueFalse。然后,使用该数组作为索引即可选择数组中对应位置为 True 的元素。

例如,假设有一个 numpy 数组 a,其中包含了一些整数:

import numpy as np

a = np.array([1, 2, 3, 4, 5])

现在,我们想要选择所有大于 2 的元素。可以通过以下步骤实现:

  1. 创建一个布尔型数组,其中数组的每个元素都是 TrueFalse。可以使用比较运算符(例如 ><== 等)来创建布尔型数组:
    mask = a > 2
    

  2. 使用布尔型数组 mask 作为索引来选择数组中对应位置为 True 的元素:  
    b = a[mask] # 选择大于 2 的元素
    

 3、选择‘False’

b = a[~mask] # 选择小于 2 的元素

猜你喜欢

转载自blog.csdn.net/m0_62865498/article/details/130531473
今日推荐