NumPy中怎么拆分数组

numpy

631

2024/5/20 18:25:50

栏目: 编程语言

NumPy中可以使用split函数来拆分数组，语法如下：

numpy.split(ary, indices_or_sections, axis=0)

其中，ary表示要拆分的数组，indices_or_sections可以是一个整数或者一个数组，用于指定拆分的位置或者拆分的个数，axis表示沿着哪个轴进行拆分。

例如，将一个一维数组拆分成多个子数组：

import numpy as np

arr = np.array([1, 2, 3, 4, 5, 6])
subarrays = np.split(arr, 3)

print(subarrays)
# [array([1, 2]), array([3, 4]), array([5, 6])]

将一个二维数组按行拆分成多个子数组：

arr = np.array([[1, 2], [3, 4], [5, 6]])
subarrays = np.split(arr, 3)

print(subarrays)
# [array([[1, 2]]), array([[3, 4]]), array([[5, 6]])]

需要注意的是，拆分后得到的子数组是视图而不是副本，修改子数组会影响原始数组。

最新知识库