import numpy as np
import random
a=np.zeros((2,4))
#函数:用于将多维数组左右拆分成两份
def DvdArr(arr,n):
x1=arr[:,0,]
for i in range(len(a)):
for j in range(len(a[0])):
a[i][j]=random.random()
print("原a数组:")
print(a)
b11=a[:,0:2]#按列切分,从第1列切分到第2列
print("按列切分b11:")
print(b11)
b21=a[0:1,:]#按行切分,从第1列切分到第1列
print("按行切分b21:")
print(b21)
输出:
原a数组:
[[0.1382644 0.77630067 0.77940432 0.67616644]
[0.98751536 0.06283405 0.64940781 0.5341892 ]]
按列切分b11:
[[0.1382644 0.77630067]
[0.98751536 0.06283405]]
按行切分b21:
[[0.1382644 0.77630067 0.77940432 0.67616644]]