NumPy中怎么分割数组
在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进行分割,得到了两个子数组。
版权声明
本文仅代表作者观点,不代表米安网络立场。
发表评论:
◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。