以下是关于“Python 实现在多维数组中挑选符合条件的全部元素”的完整攻略,其中包含两个示例说明。
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
在本示例中,我们创建了一个 3x3 的多维数组。
result = arr[arr > 5]
在本示例中,我们使用 NumPy 库的条件索引功能,挑选出数组中大于 5 的所有元素。
arr = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
在本示例中,我们创建了一个 3x3 的多维数组。
result = [element for row in arr for element in row if element > 5]
在本示例中,我们使用 Python 原生语法,通过列表推导式挑选出数组中大于 5 的所有元素。
通过以上步骤,我们可以使用 NumPy 库或 Python 原生语法实现在多维数组中挑选符合条件的全部元素,并成功地实现了两个示例。
本文链接:http://task.lmcjl.com/news/5268.html