python&numpy十： 数组比较和布尔操作

在NumPy中，你可以使用数组比较和布尔操作来执行逻辑运算、条件筛选等操作。下面我会介绍几种常用的方法。

1.逻辑运算：

• NumPy提供了一些逻辑运算函数，如np.logical_and()、np.logical_or()和np.logical_not()等。
• 这些函数用于对两个或多个数组进行逐元素的逻辑运算，并返回一个新的布尔类型数组。

示例代码：

import numpy as np

arr1 = np.array([True, True, False, False])
arr2 = np.array([True, False, True, False])

print(np.logical_and(arr1, arr2))  # 输出: [ True False False False]
print(np.logical_or(arr1, arr2))   # 输出: [ True  True  True False]
print(np.logical_not(arr1))        # 输出: [False False  True  True]

2.条件筛选：

• 使用条件表达式可以创建一个布尔类型数组，用于表示数组中每个元素是否满足特定条件。
• 可以使用这个布尔类型数组对原始数组进行索引操作，从而获取满足条件的子集。

示例代码：

import numpy as np

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

# 获取大于2的元素
condition = arr > 2
filtered_arr = arr[condition]
print(filtered_arr)  # 输出: [3 4 5]

# 将小于等于3的元素设置为0
arr[arr <= 3] = 0
print(arr)  # 输出: [0 0 0 4 5]

这些方法可以帮助你执行数组比较和布尔操作，用于逻辑运算、条件筛选等。如果你还有其他问题，请随时向我提问！