首页数组numpy的条件筛选

数组numpy的条件筛选

时间: 2023-04-03 20:04:20 浏览: 108

从numpy数组中取出满足条件的元素示例

例如问题：从 arr 数组中提取所有奇数元素。 input：arr = np.array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9]) output: #> array([1, 3, 5, 7, 9]) Solution: #Input >>> arr = np.array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9]) #Solution arr = arr[arr % 2 == 1] >>> array([1, 3, 5, 7, 9]) 这是对奇数元素的提取，其他条件同理，在这个问题上我们深入再学习一下，首先 numpy 中的 array 数组是可以直

可以使用numpy的布尔索引来进行条件筛选，例如： import numpy as np arr = np.array([1, 2, 3, 4, 5]) condition = arr > 3 result = arr[condition] print(result) 输出结果为：[4 5]

阅读全文