У меня есть объект * Массив numpy содержит более 10000 элементов, возвращаемых из KDTree, каждый элемент является вектором 1xN (n может отличаться для каждого элемента). Каждый вектор содержит числа с плавающей точкой как вектор (например, [0. 3 12]). Я хотел бы найти индекс 0 в каждом векторе для всех элементов в этом массиве объектов.
Чтобы ускорить l oop, я использовал .tolist (), что дает мне некоторое улучшение в вычислениях скорость. Мне интересно, есть ли способ получить нулевой индекс вместо циклического прохождения каждого элемента?
например
a = [np.array([0, 1, 5]), np.array([2, 3]), np.array([1,0,5])]
[array([0, 1, 5]), array([2, 3]), array([1, 0, 5])]
Есть ли способ получить индекс ноль в? Как [[0,0], [2,2]]
Тогда у меня есть другой список numpy массив, который имеет тот же размер, что и. Я хотел бы применить нулевой индекс a, чтобы либо взять / удалить эти числа в b.
b = [np.array([2, 3, 4]), np.array([4, 5]), np.array([8,9,2])]
Так что, если есть быстрый способ получить [2,9] как один вектор (применяя нулевой индекс a к b).