Python(5) Numpy中array的切片相关操作

在交叉验证中经常需要轮流从数组中取出一些内容,此时需要对数组做一些切片以及shape, rashape的操作,总结如下:

import numpy as np

#构造数组
a = [i for i in range(100)]
a = np.array(a)
a_size = a.shape
a = np.reshape(a, (10, 10))

print('a:')
print(a)
print('#######################################################')

#数组分成5份
a_item = np.split(a, 5)
print('a_item:')
print(a_item)
print('#######################################################')

#从数组中选出相应的份
a_test = a_item[:0] + a_item[1:5]
a_verify = a_item[0]
print('a_test:')
print(a_test)
print('a_verify:')
print(a_verify)
print('#######################################################')

#将选出的份恢复成二维数组形式
a_test = np.reshape(a_test, (int(a.shape[0] * 4 / 5), -1))
a_verify = np.reshape(a_verify, (int(a.shape[0] / 5), -1))
print('恢复后的a_test:')
print(a_test)
print('恢复后的a_verify:')
print(a_verify)
print('#######################################################')



猜你喜欢

转载自blog.csdn.net/BoRenLiang/article/details/79586274