np.array_split()与np.split()的区别

最新推荐文章于 2024-04-23 05:30:00 发布

际遇98

最新推荐文章于 2024-04-23 05:30:00 发布

阅读量550

点赞数

分类专栏： python基础篇

python基础篇专栏收录该内容

43 篇文章 0 订阅

订阅专栏

本文转自 https://blog.csdn.net/m0_37393514/article/details/79537639

np.split():

大小要么按照数字来划分（int），要么是一个list来划分。但是如果你仅是输入一个int类型的数字的话，你的数组必须是均等的分割，否则就会报错，不像array_split()是可以进行不均等的划分的！

x = np.arange(9.0)
np.split(x, 3)
[array([ 0.,  1.,  2.]), array([ 3.,  4.,  5.]), array([ 6.,  7.,  8.])]
 
x = np.arange(8.0)
np.split(x, [3, 5, 6, 10])
[array([ 0.,  1.,  2.]),
array([ 3.,  4.]),
array([ 5.]),
array([ 6.,  7.]),
array([], dtype=float64)]

np.array_split():

x = np.arange(8.0)
np.array_split(x, 3)
[array([ 0.,  1.,  2.]), array([ 3.,  4.,  5.]), array([ 6.,  7.])]
 
x = np.arange(7.0)
np.array_split(x, 3)
[array([ 0.,  1.,  2.]), array([ 3.,  4.]), array([ 5.,  6.])]