NumPy 逻辑运算|极客教程

NumPy 逻辑运算

在数据处理和分析中，逻辑运算是非常重要的一部分。在Python中，我们通常使用NumPy库来进行高效的数组运算。NumPy提供了丰富的逻辑运算函数，可以帮助我们对数组进行逻辑运算，比较元素之间的大小关系，筛选符合条件的元素等。本文将详细介绍NumPy中的逻辑运算函数，并提供示例代码来帮助读者更好地理解和应用这些函数。

1. 逻辑运算函数

1.1 np.logical_and()

np.logical_and()函数用于对两个数组进行逻辑与操作，返回一个新的数组，数组中的每个元素都是对应位置上两个数组元素的逻辑与结果。

import numpy as np

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

result = np.logical_and(arr1, arr2)
print(result)

Output:

NumPy 逻辑运算

1.2 np.logical_or()

np.logical_or()函数用于对两个数组进行逻辑或操作，返回一个新的数组，数组中的每个元素都是对应位置上两个数组元素的逻辑或结果。

import numpy as np

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

result = np.logical_or(arr1, arr2)
print(result)

Output:

NumPy 逻辑运算

1.3 np.logical_not()

np.logical_not()函数用于对数组进行逻辑非操作，返回一个新的数组，数组中的每个元素都是原数组元素的逻辑非结果。

import numpy as np

arr = np.array([True, False, True, False])

result = np.logical_not(arr)
print(result)

Output:

NumPy 逻辑运算

1.4 np.logical_xor()

np.logical_xor()函数用于对两个数组进行逻辑异或操作，返回一个新的数组，数组中的每个元素都是对应位置上两个数组元素的逻辑异或结果。

import numpy as np

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

result = np.logical_xor(arr1, arr2)
print(result)

Output:

NumPy 逻辑运算

2. 比较运算函数

2.1 np.greater()

np.greater()函数用于对两个数组进行逐元素的大于比较，返回一个新的数组，数组中的每个元素都是对应位置上两个数组元素的比较结果。

import numpy as np

arr1 = np.array([1, 2, 3, 4])
arr2 = np.array([2, 2, 3, 3])

result = np.greater(arr1, arr2)
print(result)

Output:

NumPy 逻辑运算

2.2 np.greater_equal()

np.greater_equal()函数用于对两个数组进行逐元素的大于等于比较，返回一个新的数组，数组中的每个元素都是对应位置上两个数组元素的比较结果。

import numpy as np

arr1 = np.array([1, 2, 3, 4])
arr2 = np.array([2, 2, 3, 3])

result = np.greater_equal(arr1, arr2)
print(result)

Output:

NumPy 逻辑运算

2.3 np.less()

np.less()函数用于对两个数组进行逐元素的小于比较，返回一个新的数组，数组中的每个元素都是对应位置上两个数组元素的比较结果。

import numpy as np

arr1 = np.array([1, 2, 3, 4])
arr2 = np.array([2, 2, 3, 3])

result = np.less(arr1, arr2)
print(result)

Output:

NumPy 逻辑运算

2.4 np.less_equal()

np.less_equal()函数用于对两个数组进行逐元素的小于等于比较，返回一个新的数组，数组中的每个元素都是对应位置上两个数组元素的比较结果。

import numpy as np

arr1 = np.array([1, 2, 3, 4])
arr2 = np.array([2, 2, 3, 3])

result = np.less_equal(arr1, arr2)
print(result)

Output:

NumPy 逻辑运算

3. 数组元素筛选

3.1 np.where()

np.where()函数用于根据指定条件从数组中选择元素，返回一个新的数组，数组中的元素是满足条件的原数组元素。

import numpy as np

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

result = np.where(arr > 2)
print(result)

Output:

NumPy 逻辑运算

3.2 np.extract()

np.extract()函数用于根据指定条件从数组中提取元素，返回一个新的数组，数组中的元素是满足条件的原数组元素。

import numpy as np

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

result = np.extract(arr > 2, arr)
print(result)

Output:

NumPy 逻辑运算

4. 逻辑运算的应用

逻辑运算在数据处理和分析中有着广泛的应用，可以帮助我们筛选数据、过滤异常值、进行条件判断等操作。下面我们通过一个示例来演示逻辑运算的应用。

4.1 筛选数据

假设我们有一个包含成绩数据的数组，我们想要筛选出大于等于60分的成绩。

import numpy as np

scores = np.array([85, 72, 45, 60, 90, 55])

passed_scores = scores[scores >= 60]
print(passed_scores)

Output:

NumPy 逻辑运算

4.2 判断异常值

假设我们有一个包含身高数据的数组，我们想要判断哪些身高数据是异常值（小于150cm或大于200cm）。

import numpy as np

heights = np.array([160, 170, 140, 180, 190, 210])

abnormal_heights = heights[(heights < 150) | (heights > 200)]
print(abnormal_heights)

Output:

NumPy 逻辑运算

4.3 条件判断

假设我们有两个数组，分别表示学生的数学成绩和英语成绩，我们想要判断哪些学生同时数学成绩和英语成绩都大于80分的情况。

import numpy as np

math_scores = np.array([85, 72, 90, 65, 88])
english_scores = np.array([80, 75, 92, 70, 85])

high_scores_students = np.logical_and(math_scores > 80, english_scores > 80)
print(high_scores_students)