numpy.split() to funkcja, która dzieli jeden lub więcej wymiarów tablicy NumPy na kilka podtablic. Możesz użyć tej funkcji, aby podzielić tablicę na równe fragmenty lub według określonego indeksu.
Przykład użycia:
import numpy as np
# Dzielimy 1-wymiarową tablicę na dwie podtablice
x = np.array([1, 2, 3, 4, 5, 6])
A, B = np.split(x, 2)
print(A) # [1, 2, 3]
print(B) # [4, 5, 6]
# Dzielimy 2-wymiarową tablicę na dwie podtablice poziomo (wzdłuż pierwszego wymiaru)
y = np.array([[1, 2], [3, 4], [5, 6], [7, 8])
A, B = np.split(y, 2, axis=0)
print(A) # [[1, 2], [3, 4]]
print(B) # [[5, 6], [7, 8]]
# Dzielimy 2-wymiarową tablicę na dwie podtablice pionowo (wzdłuż drugiego wymiaru)
A, B = np.split(y, 2, axis=1)
print(A) # [[1], [3], [5], [7]]
print(B) # [[2], [4], [6], [8]]
# Dzielimy tablicę na trzy podtablice według indeksów
A, B, C = np.split(x, [2, 4])
print(A) # [1, 2]
print(B) # [3, 4]
print(C) # [5, 6]
Funkcja numpy.split() zwraca listę podtablic, które są wynikiem podziału. Możesz również użyć funkcji numpy.array_split(), która działa podobnie, ale zwraca tablicę zawierającą podtablice, nawet jeśli podtablice nie są równej długości.