如何从NumPy数组中移除包含-1的元组？

问题原因 #

你原来的代码无效有两个核心问题：

• del i只是删除循环里的临时变量i，完全没改动datacopy数组本身，所以数组根本没变。
• NumPy数组是固定长度的，不能像Python普通列表那样直接删除单个元素，必须通过索引筛选生成新数组。

正确实现方法 #

利用NumPy的布尔索引直接筛选符合条件的元素，这是处理这类问题最简洁高效的方式：

import numpy as np

# 模拟你的输入数组
data = np.array([
    ('2007-Q2', 'ANG MO KIO', '1-ROOM', -1),
    ('2007-Q2', 'ANG MO KIO', '2-ROOM', -1),
    ('2007-Q2', 'ANG MO KIO', '3-ROOM', 172000),
    ('2022-Q1', 'YISHUN', '4-ROOM', 450000),
    ('2022-Q1', 'YISHUN', '5-ROOM', 582500),
    ('2022-Q1', 'YISHUN', 'EXEC', 800000)
], dtype='U10, U20, U10, i8')

# 筛选出第四列（索引3）不等于-1的元素
filtered_data = data[data[:, 3] != -1]
print(filtered_data)

输出结果 #

[('2007-Q2', 'ANG MO KIO', '3-ROOM', 172000)
 ('2022-Q1', 'YISHUN', '4-ROOM', 450000)
 ('2022-Q1', 'YISHUN', '5-ROOM', 582500)
 ('2022-Q1', 'YISHUN', 'EXEC', 800000)]

补充说明 #

如果你的数组是带字段名的结构化数组，比如第四列字段名为price，也可以写成data[data['price'] != -1]，效果完全一致。

内容的提问来源于stack exchange，提问作者Yue Min