NumPy中分割數組的幾種常見用法，你知道幾種？
作者：Hank Zheng
我們使用np.hsplit?函數將數組?arr?水平分割成兩個子數組，每個子數組包含相同數量的列。使用?np.vsplit?函數將數組?arr?垂直分割成兩個子數組，每個子數組包含相同數量的行。
在NumPy中，分割（split）數組是一種常見的操作，它允許我們將一個大的數組分割成更小的子數組。這在數據處理和分析中非常有用。本文將介紹NumPy中分割數組的幾種常見用法，并提供相應的代碼示例。
首先，讓我們導入NumPy庫：
import numpy as np- 均等分割
arr = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
# 將數組均等分割成兩個子數組
result = np.split(arr, 2)
print(result)輸出：
[array([1, 2, 3, 4, 5]), array([6, 7, 8, 9, 10])]在這個例子中，我們使用np.split函數將數組arr均等地分割成兩個子數組。分割后的結果是一個包含兩個子數組的列表。
- 不等分割
arr = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
# 按照指定的索引位置分割數組
result = np.split(arr, [2, 5, 7])
print(result)輸出：
[array([1, 2]), array([3, 4, 5]), array([6, 7]), array([8, 9, 10])]在這個例子中，我們使用np.split函數根據指定的索引位置將數組arr進行分割。在索引位置2、5和7處進行分割，得到四個子數組。
- 按行分割多維數組
arr = np.array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9],
[10, 11, 12]])
# 按行分割多維數組
result = np.split(arr, 2, axis=0)
print(result)輸出：
[array([[1, 2, 3],
[4, 5, 6]]), array([[ 7, 8, 9],
[10, 11, 12]])]在這個例子中，我們使用np.split函數按行分割多維數組arr。通過指定axis=0，我們告訴函數在行的維度上進行分割。結果是一個包含兩個子數組的列表。
- 按列分割多維數組
arr = np.array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9],
[10, 11, 12]])
# 按列分割多維數組
result = np.split(arr, 3, axis=1)
print(result)輸出：
[array([[ 1],
[ 4],
[ 7],
[10]]), array([[ 2],
[ 5],
[ 8],
[11]]), array([[ 3],
[ 6],
[ 9],
[12]])]在這個例子中，我們使用np.split函數按列分割多維數組arr。通過指定axis=1，我們告訴函數在列的維度上進行分割。結果是一個包含三個子數組的列表。
- 水平分割和垂直分割
arr = np.arange(16).reshape((4, 4))
# 水平分割數組
result_horizontal = np.hsplit(arr, 2)
# 垂直分割數組
result_vertical = np.vsplit(arr, 2)
print(result_horizontal)
print(result_vertical)輸出：
[array([[ 0, 1],
[ 4, 5],
[ 8, 9],
[12, 13]]), array([[ 2, 3],
[ 6, 7],
[10, 11],
[14, 15]])]
[array([[0, 1, 2, 3],
[4, 5, 6, 7]]), array([[ 8, 9, 10, 11],
[12, 13, 14, 15]])]在這個例子中，我們使用np.hsplit函數將數組arr水平分割成兩個子數組，每個子數組包含相同數量的列。使用np.vsplit函數將數組arr垂直分割成兩個子數組，每個子數組包含相同數量的行。
這些是NumPy中分割數組的幾種常見用法。無論是均等分割還是不等分割，以及多維數組的行分割和列分割，NumPy提供了靈活的函數來滿足不同的分割需求。通過熟練掌握這些技巧，您可以更有效地處理和分析大型數據集。
責任編輯：武曉燕
來源：
科學隨想錄
