python - 如何在 numpy 数组中找到元组的索引？

Question

我有一个 numpy 数组:

groups=np.array([('Species1',), ('Species2', 'Species3')], dtype=object)。

当我询问 np.where(groups == ('Species2', 'Species3')) 甚至 np.where(groups == groups[1]) 我得到一个空回复:(array([], dtype=int64),)

为什么会这样，我怎样才能得到这样一个元素的索引？

Answer 1

这里的问题可能是 array.__contains__() 的实现方式。参见 here .基本上问题是

print(('Species2', 'Species3') in groups)

打印错误。如果您仍然想使用 numpy.where 函数，而不是像另一个答案所建议的那样使用 for 循环，那么最好以某种方式构造一个合适的真值掩码。例如

x = np.array(list(map(lambda x: x== ('Species2', 'Species3'), groups)))
print(np.where(x))

给出正确的结果。不过，可能还有更优雅的方法。