NumPy中怎么分割數(shù)組

小億
91
2024-05-11 18:26:51

在NumPy中，可以使用numpy.split()函數(shù)來(lái)分割數(shù)組。該函數(shù)接受三個(gè)參數(shù)，分別為待分割的數(shù)組、分割點(diǎn)或分割的位置、沿著哪個(gè)軸進(jìn)行分割。

示例如下：

import numpy as np

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

# 將數(shù)組arr分割成兩部分
splitted_arr = np.split(arr, [3])

print(splitted_arr)

輸出結(jié)果為：

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

在上面的示例中，我們將數(shù)組arr按照索引3進(jìn)行分割，得到了兩個(gè)子數(shù)組。

0 踩