在数据分析和处理中，我们经常需要根据特定条件筛选数组中的元素。一个常见的需求是，根据元素与其“右侧邻居”（即后继元素）的关系来筛选。具体来说，我们可能需要找出所有这样的元素：其后继元素比它自身大至少一个预设的阈值。
例如，给定一个NumPy数组：
ex_arr = np.array([1, 2, 3, 8, 9, 10, 12, 16, 17, 23])
我们的目标是筛选出所有满足条件的元素，即 ex_arr[i+1] - ex_arr[i] >= 3。 对于上述 ex_arr，期望得到的筛选结果是：
desired_arr = [3, 12, 17]
这是因为：
NumPy提供了一个非常实用的函数 np.diff，它能够计算数组中相邻元素的差值。np.diff(arr) 返回一个新数组，其中 result[i] = arr[i+1] - arr[i]。值得注意的是，np.diff 的输出数组长度会比原始数组少一个元素。
例如：
import numpy as np ex_arr = np.array([1, 2, 3, 8, 9, 10, 12, 16, 17, 23]) diff_arr = np.diff(ex_arr) print(diff_arr) # 输出: [1 1 5 1 1 2 4 1 6]
这里的 diff_arr[0] (1) 对应 ex_arr[1] - ex_arr[0] (2-1)，diff_arr[1] (1) 对应 ex_arr[2] - ex_arr[1] (3-2)，以此类推。diff_arr 的最后一个元素 (6) 对应 ex_arr[9] - ex_arr[8] (23-17)。
这种方法的核心思想是先利用 np.diff 生成一个布尔数组，表示相邻元素差值是否满足条件，然后通过填充 False 来匹配原始数组的长度，最终进行布尔索引。
步骤：
示例代码：
import numpy as np
ex_arr = np.array([1, 2, 3, 8, 9, 10, 12, 16, 17, 23])
# 1. 计算相邻差值并应用条件
condition_mask = (np.diff(ex_arr) >= 3)
print("原始条件掩码:", condition_mask)
# 输出: 原始条件掩码: [False False True False False False True False True]
# 2. 使用 np.r_ 扩展掩码，在末尾添加 False
# np.r_ 将多个数组或标量按行连接起来
full_mask = np.r_[condition_mask, False]
print("扩展后的掩码:", full_mask)
# 输出: 扩展后的掩码: [False False True False False False True False True False]
# 3. 使用布尔索引筛选数组
desired_arr = ex_arr[full_mask]
print("筛选结果:", desired_arr)
# 输出: 筛选结果: [ 3 12 17]工作原理分析：np.diff(ex_arr) 得到的是 [ex_arr[1]-ex_arr[0], ex_arr[2]-ex_arr[1], ..., ex_arr[N-1]-ex_arr[N-2]]。 当 np.diff(ex_arr)[i] >= 3 为 True 时，意味着 ex_arr[i+1] - ex_arr[i] >= 3。此时，我们希望选择的是 ex_arr[i]。 因此，condition_mask[i] 对应的是 ex_arr[i] 是否应该被选择。 由于 condition_mask 比 ex_arr 短一个元素，condition_mask[0] 对应 ex_arr[0]，condition_mask[N-2] 对应 ex_arr[N-2]。 ex_arr[N-1]（最后一个元素）没有任何后继元素，所以它不可能满足条件，其对应的布尔值应为 False。通过 np.r_[condition_mask, False]，我们为 ex_arr[N-1] 补上了 False，使得掩码长度与 ex_arr 匹配，从而实现正确的布尔索引。
另一种简洁的方法是使用 np.nonzero 函数。np.nonzero(arr) 返回一个元组，其中包含数组中非零元素的索引。当用于布尔数组时，它返回 True 元素的索引。
步骤：
示例代码：
import numpy as np
ex_arr = np.array([1, 2, 3, 8, 9, 10, 12, 16, 17, 23])
# 1. 计算相邻差值并应用条件
condition_mask = (np.diff(ex_arr) >= 3)
print("条件掩码:", condition_mask)
# 输出: 条件掩码: [False False True False False False True False True]
# 2. 使用 np.nonzero 获取满足条件的索引
# np.nonzero 返回一个元组，我们需要取出第一个元素（索引数组）
indices = np.nonzero(condition_mask)[0]
print("满足条件的索引:", indices)
# 输出: 满足条件的索引: [2 6 8]
# 3. 使用整数数组索引筛选数组
desired_arr = ex_arr[indices]
print("筛选结果:", desired_arr)
# 输出: 筛选结果: [ 3 12 17]工作原理分析：np.diff(ex_arr) >= 3 得到的布尔数组 condition_mask 中，True 值所在的索引 i 意味着 ex_arr[i+1] - ex_arr[i] >= 3。 我们希望选择的是 ex_arr[i]。np.nonzero(condition_mask)[0] 正好返回了这些 i 值。 例如，当 condition_mask[2] 为 True 时，np.nonzero 会返回 2。这个 2 正好是 ex_arr 中 3 的索引，而 3 就是我们希望选中的元素（因为 ex_arr[3]-ex_arr[2] 即 8-3=5 满足条件）。 这种方法避免了手动调整掩码长度，因为它直接操作的是索引，而 np.diff 的长度特性恰好与我们对“前一个元素”的筛选需求相匹配。
本教程介绍了在NumPy中高效筛选数组的两种方法，其核心在于利用 np.diff 函数计算相邻元素的差值，并结合布尔掩码或 np.nonzero 进行索引。这两种方法都体现了NumPy矢量化操作的强大之处，能够以简洁高效的方式解决基于元素与后继元素关系的复杂筛选问题。掌握这些技巧将有助于提升您在Python中进行数据处理的效率和代码质量。
以上就是使用NumPy高效筛选数组：基于与后继元素的差值条件的详细内容，更多请关注php中文网其它相关文章！
每个人都需要一台速度更快、更稳定的 PC。随着时间的推移，垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是，许多工具可以让 Windows 保持平稳运行。
Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号
682
收藏
