Python | Numpy.dsplit() 方法

在numpy中，dsplit()方法用于按指定轴拆分一个数组为多个子数组。

语法

numpy.dsplit(arr, indices_or_sections)

参数说明

• arr：要拆分的数组
• indices_or_sections：拆分时使用的索引或者分割点

返回值

返回拆分后的子数组构成的列表

示例

import numpy as np

arr = np.arange(16).reshape(4, 4)
print(arr)

输出：

[[ 0  1  2  3]
 [ 4  5  6  7]
 [ 8  9 10 11]
 [12 13 14 15]]

result = np.dsplit(arr, 2)
print(result[0])

输出：

[[ 0  1]
 [ 4  5]
 [ 8  9]
 [12 13]]

解释

在上述示例中，先创建一个4x4的二维数组，然后使用dsplit()方法将数组按指定轴拆分，这里按照第二维度将数组拆分为两个子数组，结果会被存储在一个列表中，这个列表的第一个元素就是我们想要的第一个子数组。