Курс Python → Numpy: разбиение массивов

Библиотека Numpy предоставляет мощные инструменты для работы с массивами данных в Python. Одним из часто используемых методов является разбиение массивов. В прошлом посте мы уже рассмотрели способы объединения массивов, и теперь настало время узнать, как разделить массивы на части.

Для более гибкого разделения массивов в Numpy представлены специальные методы: hsplit, vsplit и array_split. Метод hsplit разбивает массив вдоль горизонтальной оси, то есть по строкам, а vsplit — вдоль вертикальной оси, то есть по столбцам. Эти методы позволяют эффективно разбивать массивы на части без необходимости использования сложных срезов.

Однако, если вам нужно разделить массив по другой оси или произвольным образом, то вам может помочь метод array_split. Он является более общим и позволяет указать вдоль какой оси произойдет разбиение массива. Этот метод предоставляет большую гибкость и контроль над процессом разбиения массива.

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# Разбиение массива по горизонтальной оси
parts = np.hsplit(arr, 3)

# Разбиение массива по вертикальной оси
parts = np.vsplit(arr, 3)

# Общее разбиение массива
parts = np.array_split(arr, 2, axis=1)

Приведенный выше код демонстрирует использование методов hsplit, vsplit и array_split для разбиения массива на части вдоль различных осей. Используйте эти методы в зависимости от ваших потребностей при работе с массивами данных в Numpy.

Твои коллеги будут рады, поделись в