Удалите последовательный индекс массива NumPy в цикле над его содержимым

Question

Я повторяю массив NumPy (nd) строк OpenCV. Я хочу удалить все строки с 8-градусной вертикалью. Я понимаю, что массив numpy неизменен, а то, что я делаю в коде, неверно, но демонстрирует идею того, что я пытаюсь сделать;

index = 0 
for line in self.lines[0]: 
    if (line[1]*180)/np.pi > 8:          
     self.lines[0] = np.delete(self.lines[0], index, axis=0) 
    index+=1 

Как я могу удалить эти индексы массива NumPy?

Спасибо!

Answer 1

Вы не можете удалять индексы массива в то время как итерация массива. Это даст неправильные результаты. Предположим, что вы находитесь на итерации 5, и этот индекс удовлетворяет условию if и его нужно удалить, но если мы удалим его, он заставит элемент массива в индексе 6 прийти к индексу 5, а следующий выбранный элемент будет равен 7, который приходит к индексу 6 после удаления и, следовательно, условие никогда не проверяется для элемента изначально по индексу 6, т.е. перед удалением.

Основная идея решения этой проблемы заключается в том, чтобы добавить эти индексы в список и удалить их вне очереди. Таким образом, ваш код:

index = 0 
idx = [] 
for line in self.lines[0]: 
    if (line[1]*180)/np.pi > 8:  
     idx.append(index)   
    index+=1 

self.lines[0] = np.delete(self.lines[0], idx, axis=0) 

Answer 2

Я думаю, вам не хватает отступы в коде, так что я его интерпретировать так:

index = 0 
for line in self.lines[0]: 
    if (line[1]*180)/np.pi > 8:          
     self.lines[0] = np.delete(self.lines[0], index, axis=0) 
    index+=1 

Вы можете сделать это, конечно, без петли. Сначала мы делаем проверку, чтобы построить boolean indexing array (True/False значения для каждого элемента массива):

index = self.lines[0,:,1]*180/np.pi > 8 

Затем мы выбираем строки, где условие ложно:

a[~a%2==0] 

Где np.nonzero преобразует из булевого массива в массив, где перечислены все индексы с истинными значениями.

Например:

>>> a = np.arange(10) 
>>> a[~a%2==0] 
array([1, 3, 5, 7, 9])