从numpy数组对应的列表中删除元素

在处理 NumPy 数组和 Python 列表时，有时需要根据 NumPy 数组的值来删除对应列表中的元素。假设你有一个 NumPy 数组和一个对应的列表，你希望删除 NumPy 数组中满足某些条件的元素，并同时删除对应列表中的元素。

以下是一个示例，展示如何实现这一点：

示例

假设你有一个 NumPy 数组 arr 和一个对应的列表 lst，你希望删除 arr 中所有大于某个值的元素，并同时删除 lst 中对应的元素。

import numpy as np

# 示例数据
arr = np.array([1, 2, 3, 4, 5])
lst = ['a', 'b', 'c', 'd', 'e']

# 条件：删除大于 3 的元素
condition = arr > 3

# 删除 NumPy 数组中满足条件的元素
arr = arr[~condition]

# 删除列表中对应的元素
lst = [item for i, item in enumerate(lst) if not condition[i]]

print("Updated array:", arr)
print("Updated list:", lst)

解释

1. 定义示例数据：
   • arr 是一个 NumPy 数组。
   • lst 是一个对应的 Python 列表。
2. 定义条件：
   • condition 是一个布尔数组，表示 arr 中哪些元素满足条件（大于 3）。
3. 删除 NumPy 数组中满足条件的元素：
   • 使用布尔索引 ~condition 来选择不满足条件的元素。
4. 删除列表中对应的元素：
   • 使用列表推导式和 enumerate 函数来遍历 lst，并根据 condition 中的布尔值来决定是否保留每个元素。

结果

运行上述代码后，输出将是：

Updated array: [1 2 3]
Updated list: ['a', 'b', 'c']