numpy 中 newaxis函数的使用,numpy中ravel()、flatten()、squeeze()的用法与区别
程序员文章站
2024-03-02 22:03:46
...
newaxis表示增加一个新的坐标轴
import numpy as np
a = np.array([1,2,3])
print (a.shape,'\n',a)
结果为:
(3,)
[1 2 3]
a = np.array([1,2,3])[:,np.newaxis]
print (a.shape,'\n',a)
(3, 1)
[[1]
[2]
[3]]
你会发现,和第一个程序相比,a的shape为(3,)现在为(3,1)变为二维数组了,之前为[1,2,3],现在变为 [[1]
[2]
[3]]
a = np.array([1,2,3])[np.newaxis,:]
print (a.shape,'\n',a)
输出结果为:
(1, 3)
[[1 2 3]]
这个和第二个相比,好像和他是反的,相当于转置了,这是因为和[np.newaxis,:]
这个地方np.newaxis放的位置有关,第二个程序放在[:,]的后面,相当于在原来的后面增加一个维度,所以变为(3,1),而第三个则放在前面,则为(1,3),记得注意啊,放在前面是先逗号,在冒号,而放在后面是先冒号在逗号,不要弄错了哦,同时记得是中括号扩起来,不是小括号哦
总结
np.newaxis的作用就是在原来的数组上增加一个维度。
numpy中的ravel()、flatten()、squeeze()的用法与区别
numpy中的ravel()、flatten()、squeeze()都有将多维数组转换为一维数组的功能,区别:ravel()
:如果没有必要,不会产生源数据的副本flatten()
:返回源数据的副本squeeze()
:只能对维数为1的维度降维
numpy.ravel(a, order='C')
numpy.ravel() vs numpy.flatten()首先声明两者所要实现的功能是一致的(将多维数组降位一维),两者的区别在于返回拷贝(copy)还是返回视图(view),numpy.flatten()返回一份拷贝,对拷贝所做的修改不会影响(reflects)原始矩阵,而numpy.ravel()返回的是视图(view,也颇有几分C/C++引用reference的意味),会影响(reflects)原始矩阵。