Как удалить элемент на последней оси?

Question

Как удалить элемент на последней оси?

Милана Ким Ученик (74), на голосовании 1 месяц назад

Здравствуйте, помогите, пожалуйста, я не понимаю как в numpy с помощью delete удалить элементы на последней оси. При том, что мы не знаем сколько осей. Но знаем индексы элементов, которые надо удалить.

Answer 1

 import numpy as np 
 
# Создаем тестовый массив 
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]]) 
print("Исходный массив:") 
print(arr) 
 
# Индексы элементов, которые нужно удалить на последней оси 
indices_to_delete = [1] 
 
# Получаем размерность массива 
ndim = arr.ndim 
 
# Удаляем элементы на последней оси 
new_arr = np.delete(arr, indices_to_delete, axis=ndim-1) 
 
print("\nМассив после удаления:") 
print(new_arr)