Как выполнить разделение массива NumPy?

@nelle 

В NumPy вы можете выполнить разделение массива с помощью функции split(). Эта функция позволяет разбить массив на несколько подмассивов по указанной оси.

Вот пример использования функции split():

1
2
3
4
5
6
7
8
9

import numpy as np

# создаем массив
arr = np.array([1, 2, 3, 4, 5, 6])

# разделяем массив на три равные части
result = np.split(arr, 3)

print(result)

В этом примере мы создаем одномерный массив arr и используем функцию split() для разделения массива на три части. Функция возвращает список из трех подмассивов.

Если массив не делится на равные части, то вы можете указать точки разделения явно. Например:

1
2
3
4
5
6
7
8
9

import numpy as np

# создаем массив
arr = np.array([1, 2, 3, 4, 5, 6, 7])

# разделяем массив на три части
result = np.split(arr, [2, 5])

print(result)

В этом примере мы создаем одномерный массив arr и используем функцию split() для разделения массива на три части. Мы указываем точки разделения явно с помощью списка [2, 5], который содержит индексы элементов, где нужно выполнить разделение. Функция возвращает список из трех подмассивов, которые соответствуют этим точкам разделения.