np.split(x, [ ])用法

最新推荐文章于 2024-05-26 10:49:06 发布

原创最新推荐文章于 2024-05-26 10:49:06 发布 · 2.5k 阅读

5 ·

CC 4.0 BY-SA版权

文章标签：

#np.split #split

np.split(x, [3, 5, 6, 10])

split(ary, indices_or_sections, axis=0) :把一个数组从左到右按顺序切分
#ary:要切分的数组
#indices_or_sections:如果是一个整数，就用该数平均切分，如果是一个数组，为沿轴切分的位置（左开右闭）
#axis：沿着哪个维度进行切向，默认为0，横向切分。为1时，纵向切分

import numpy as np
x = np.arange(8)
y = np.split(x,[3,4,6,10])
print('y = ',y)
y1 = np.split(x,[3,5,6,10])
print('y1 = ',y1)
y2 = np.split(x,[3,5,6])
print('y2 = ',y2)
y3 = np.split(x,[3,5,6,7])
print('y3 = ',y3)

---------------------------------------

y = [array([0, 1, 2]), array([3]), array([4, 5]), array([6, 7]), array([], dtype=int32)]

y1 = [array([0, 1, 2]), array([3, 4]), array([5]), array([6, 7]), array([], dtype=int32)]
y2 = [array([0, 1, 2]), array([3, 4]), array([5]), array([6, 7])]
y3 = [array([0, 1, 2]), array([3, 4]), array([5]), array([6]), array([7])]