如何有效地檢查有序的 numpy 數組是否包含值？

Question

我有一個二維 numpy 數組，我知道它是跨軸 = 1 排序的。 例子：

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

如果數組包含某一行，我想檢查 log(n) 的復雜性，例如：[0,7,1]。

我考慮過使用搜索排序來獲取該行必須在 go 中的索引，如果該行確實在數組中，它必須在該索引之前或之后，但我可以確定它會是一種方式還是另一種方式？

Answer 1

您可以創建一個返回索引的 function：

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