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

Пользователь

от nelle , в категории: Python , год назад

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

Facebook Vk Ok Twitter LinkedIn Telegram Whatsapp

1 ответ

Пользователь

от narciso , год назад

@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], который содержит индексы элементов, где нужно выполнить разделение. Функция возвращает список из трех подмассивов, которые соответствуют этим точкам разделения.