记录篇：numpy相关操作

1.把numpy数组（矩阵）弄成一维的，并且可以遍历

import numpy as np
array = np.arange(4).reshape(2,2)
print("原数组：",array)

#变成一维
print("变成一维",array.flatten())

#遍历
print("遍历：")
for i in array.flat:
     print(i)

结果：

 

2.给numpy一维数组添加维度

import numpy as np
a = np.array([1,1,1])
#本身是没有维度的，即，并不是1行3列
print("输出原shape：",a.shape)

#添加维度之后，变成1行3列
a = a[np.newaxis,:]
print("添加维度之后的shape：",a.shape)

既可以在【行】上添加维度，变成1行3列，也可以在【列】上添加维度，变成3行1列。

3.np.split()函数只能在列或行上做array的均等分割，np.array_split()可以做不均等分割。

import numpy as np
a = np.arange(12).reshape(3,4)

#实现在列上的均等分割,axis=1
print(np.split(a,2,axis=1))

#实现在行上的均等分割,axis=0
print(np.split(a,3,axis=0))

#实现不均等分割，4列矩阵分割成3列
print(np.array_split(a,3,axis=1))