У меня есть матрица 2d-np.array, и я хотел бы удалить все строки, содержащие элемент x в определенном столбце. Моя цель — вернуть матрицу без этих строк, поэтому она должна быть меньше.
Моя функция выглядит так:
Код: Выделить всё
def delete_rows(matrix, x, col):
for i in range(matrix.shape[0]-1):
if(matrix[i,col] == x):
np.delete(matrix, i, axis = 0)
return matrix
К сожалению, в моем тесте форма матрицы осталась прежней после удаления строк. Я думаю, что удаленные строки были заменены строками с нулями.
Какой-нибудь совет, как мне достичь своей цели?
Подробнее здесь:
https://stackoverflow.com/questions/724 ... ic-element