温馨提示×

NumPy中怎么分割数组

小亿
103
2024-05-11 18:26:51
栏目: 编程语言
开发者测试专用服务器限时活动,0元免费领,库存有限,领完即止! 点击查看>>

在NumPy中,可以使用numpy.split()函数来分割数组。该函数接受三个参数,分别为待分割的数组、分割点或分割的位置、沿着哪个轴进行分割。

示例如下:

import numpy as np

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

# 将数组arr分割成两部分
splitted_arr = np.split(arr, [3])

print(splitted_arr)

输出结果为:

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

在上面的示例中,我们将数组arr按照索引3进行分割,得到了两个子数组。

亿速云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>

推荐阅读:python中numpy库如何进行数组分割

0