Функция vsplit

<~copy~>

Функция vsplit разделяет массив по вертикали, то есть по строкам (оси 0). Первым параметром функция принимает массив для разделения. Вторым параметром указывается число частей, на которые нужно разбить массив, либо индексы для разбиения.

Синтаксис

np.vsplit(arr, sections)

Пример

Разделим двумерный массив на две части по вертикали:

import numpy as np

arr = np.array([[1, 2, 3],
				[4, 5, 6],
				[7, 8, 9],
				[10, 11, 12]])
res = np.vsplit(arr, 2)
print(res)

Результат выполнения кода:

[array([[1, 2, 3],
		[4, 5, 6]]),
 array([[7, 8, 9],
		[10, 11, 12]])]

Пример

Разделим массив на три части по вертикали:

import numpy as np

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

Результат выполнения кода:

[array([[1, 2, 3]]),
 array([[4, 5, 6]]),
 array([[7, 8, 9]])]

Пример

Разделим массив по указанным индексам строк:

import numpy as np

arr = np.array([[1, 2, 3],
				[4, 5, 6],
				[7, 8, 9],
				[10, 11, 12],
				[13, 14, 15]])
res = np.vsplit(arr, [2, 4])
print(res)

Результат выполнения кода:

[array([[1, 2, 3],
		[4, 5, 6]]),
 array([[7, 8, 9],
		[10, 11, 12]]),
 array([[13, 14, 15]])]

Смотрите также

функцию hsplit,
которая разделяет массив по горизонтали
функцию dsplit,
которая разделяет массив по глубине
функцию split,
которая разделяет массив по указанной оси
функцию array_split,
которая разделяет массив без проверки на равные части