python - 如何有效地检查有序 numpy 数组是否包含值？

Question

我有一个二维 numpy 数组，我知道它是跨轴 = 1 排序的。例子：

[[0,0,1],
[0,7,1],
[1,1,0]]

如果数组包含某一行，我想检查 log(n) 的复杂性，例如：[0,7,1]。

我考虑过使用搜索排序来获取该行必须进入的索引，如果该行确实在数组中，它必须在该索引之前或之后，但我可以确定它会是一种方式还是另一种方式？

标签： pythonnumpysearch

Answer 1

您可以创建一个返回索引的函数：

idxlist=[[0,0,1],[0,7,1],[1,1,0]]

def checkforindex(idxlist, idx):
    return idxlist.index(idx)

print(checkforindex(idxlist, [1,1,0]))
2