数组的分割


import numpy as np

a = np.arange(9).reshape(3,3)
print('原始数组:\n',a)
b = 2 * a
print('乘2后的数组:\n',b)

'''
#(1) 水平分割 下面的代码将把数组沿着水平方向分割为3个相同大小的子数组:
c = np.hsplit(a,3)
print("分割后的数组:\n",c)
'''

'''
# 对同样的数组,调用split函数并在参数中指定参数axis=1,对比一下结果:
c = np.split(a,3,axis=1)
print("分割后的数组:\n",c)
'''

'''
# (2)垂直分割 vsplit函数将把数组沿着垂直方向分割:
c = np.vsplit(a,3)
print("分割后的数组:\n",c)

# 同样,调用split函数并在参数中指定参数axis=0,也可以得到同样的结果:
c = np.split(a,3,axis=0)
print("分割后的数组:\n",c)
'''

# (3) 深度分割 不出所料,dsplit函数将按深度方向分割数组。我们先创建一个三维数组:
c = np.arange(27).reshape(3,3,3)
d = np.dsplit(c,3)
print("分割后的数组:\n",d)

猜你喜欢

转载自www.cnblogs.com/jerryspace/p/10023242.html
今日推荐