🔎 Как без проблем удалить размерность numpy?

import numpy as np

# Создание массива с размерностью (1, 2, 1)
arr = np.array([[[1], [2]]])

# Удаление размерности
new_arr = np.squeeze(arr)

print(new_arr)
# Вывод: [1, 2]
  

Как удалить размерность numpy

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

Для начала, давайте рассмотрим, что такое размерность в NumPy. Размерность определяет количество осей в массиве данных. Например, одномерный массив имеет размерность 1, двумерный массив - размерность 2, и так далее. Иногда возникает необходимость в удалении одной или нескольких размерностей из массива данных. Давайте рассмотрим несколько примеров, как это можно сделать с помощью NumPy.

Пример 1: Удаление одной размерности

import numpy as np

# Создаем массив размерностью 2x3x4
arr = np.arange(24).reshape((2, 3, 4))
print("Исходный массив:")
print(arr)

# Удаляем одну размерность
new_arr = np.squeeze(arr)
print("Массив после удаления размерности:")
print(new_arr)

В этом примере мы создаем трехмерный массив размерностью 2x3x4 и затем используем функцию np.squeeze() для удаления одной размерности. Результатом является двумерный массив размерности 2x12.

Пример 2: Удаление нескольких размерностей

import numpy as np

# Создаем массив размерностью 1x3x1x4
arr = np.arange(12).reshape((1, 3, 1, 4))
print("Исходный массив:")
print(arr)

# Удаляем две размерности
new_arr = np.squeeze(arr, axis=(0, 2))
print("Массив после удаления размерностей:")
print(new_arr)

В этом примере мы создаем четырехмерный массив размерностью 1x3x1x4 и затем используем функцию np.squeeze() для удаления двух размерностей (осей 0 и 2). Результатом является двумерный массив размерности 3x4.

Пример 3: Удаление размерностей с условием

import numpy as np

# Создаем массив размерностью 2x3x4
arr = np.arange(24).reshape((2, 3, 4))
print("Исходный массив:")
print(arr)

# Удаляем размерности, у которых длина равна 1
new_arr = np.squeeze(arr, axis=(1,))
print("Массив после удаления размерностей:")
print(new_arr)

В этом примере мы создаем трехмерный массив размерностью 2x3x4 и затем используем функцию np.squeeze() для удаления размерности, у которой длина равна 1 (в данном случае это осями с индексами 1). Результатом является двумерный массив размерности 2x4.

Теперь вы знаете, как удалить размерность в NumPy. Эти примеры помогут вам осознать, как использовать функцию np.squeeze() для удаления размерностей из массива данных. Надеюсь, эта статья была полезной для вас!