Numpy——数组分割

最新推荐文章于 2024-04-08 05:04:13 发布

程旭员

最新推荐文章于 2024-04-08 05:04:13 发布

阅读量1.9k

点赞数 4

分类专栏： Numpy 文章标签：数组分割 Numpy 深度学习 python

本文链接：https://blog.csdn.net/weixin_37763870/article/details/104830513

版权

6 篇文章 1 订阅

订阅专栏

1.纵向等量分割

import numpy as np
A = np.arange(1,25).reshape((4,6))

print(A)
print(np.split(A, 3 ,axis = 1))  #纵向等分3列(第二个参数一定要能被总列数整除)

在这里插入图片描述

import numpy as np
A = np.arange(1,25).reshape((4,6))

print(A)
print(np.split(A, 2, axis = 0))   #横向等分2行(第二个参数一定要能被总行数整除)

在这里插入图片描述

在机器学习时经常会需要将数据做不等量的分割，因此解决办法为np.array_split()

import numpy as np
A = np.arange(1,25).reshape((4,6))

print(A)
print(np.array_split(A,4,axis = 1))  #6列分成了2、2、1、1

在这里插入图片描述

print(np.array_split(A,3,axis = 0))  #4行分成了2、1、1

在这里插入图片描述

print(np.vsplit(A,2))  #等价于print(np.split(A, 2, axis = 0))

在这里插入图片描述

print(np.hsplit(A,3))  #等价于print(np.split(A, 3 ,axis = 1))

在这里插入图片描述

关注