Как эффективно проверить, содержит ли упорядоченный массив numpy значение?

Question

У меня есть двумерный массив, который, как я знаю, отсортирован по оси = 1. Пример:

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

Я хочу со сложностью проверить log (n), содержит ли массив определенную строку, пример: [0,7,1].

Я думал об использованииотсортированный по поиску, чтобы получить индекс, в который должна войти строка, и если строка действительно находится в массиве, она должна быть до или после этого индекса, но могу ли я быть уверен, что это будет один илидругой

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