如何移除Numpy数组中与集合字符串匹配的行

错误原因分析 #

你写的代码存在两个核心问题：

1. 判断逻辑错误：item not in x是检查集合item是否包含在单个字符串元素x中（比如检查{'Honey','Grape'}是否在'Mango'里），这完全不符合需求，正确的判断应该是x not in item（检查元素是否不在目标集合中）。
2. 取值与赋值错误：你的推导式只提取了数组的第一列元素，而非整行，且直接对原数组array[:]赋值会因新、旧数组长度不一致报错。

正确解决方案 #

方法1：Numpy布尔索引（推荐，高效适配Numpy特性）

利用np.isin生成布尔掩码，直接过滤数组：

import numpy as np

# 初始化原数组与目标集合
array = np.array([['Mango', 0.75, 0.25],
                  ['Honey', 0.75, 0.25],
                  ['Grape', 0.625, 0.375],
                  ['Pineapple', 0.5, 0.5]], dtype=object)
item = {'Honey','Grape'}

# 生成掩码：保留第一列元素不在item中的行
mask = ~np.isin(array[:, 0], item)
# 过滤得到目标数组
filtered_array = array[mask]

print(filtered_array)

输出结果：

array([['Mango', 0.75, 0.25],
       ['Pineapple', 0.5, 0.5]], dtype=object)

方法2：列表推导式（直观易懂，适合小数据量）

遍历数组每一行，筛选符合条件的行后重新转为Numpy数组：

import numpy as np

array = np.array([['Mango', 0.75, 0.25],
                  ['Honey', 0.75, 0.25],
                  ['Grape', 0.625, 0.375],
                  ['Pineapple', 0.5, 0.5]], dtype=object)
item = {'Honey','Grape'}

filtered_array = np.array([row for row in array if row[0] not in item], dtype=object)

print(filtered_array)

输出结果与方法1一致。

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