小知识点,当作学习笔记记录一下。
newaxis的功能其实很简单,主要就是加入一个新的维度,大家可以参照我下面给出的例子,自行理解!
示例:
import numpy as np
x_data = np.linspace(-1,1,3)
y_data = np.linspace(-1,1,3)[:,np.newaxis]
z_data = np.linspace(-1,1,3)[np.newaxis,:]
print(x_data.shape)
print(y_data.shape)
print(z_data.shape)
print(x_data)
print(y_data)
print(z_data)
结果:
(3,)
(3, 1)
(1, 3)
[-1. 0. 1.]
[[-1.]
[ 0.]
[ 1.]]
[[-1. 0. 1.]]
注:linspace函数的功能是创建等差数列,例子中给定了三个参数,意思分别是开始元素,结束元素,生成等差数列中的元素个数。我给的例子中的意思就是从-1到1,生成具有3个元素的等差数列,所以生成的结果为[-1. 0. 1.]。
y_data为x_data在列中加了一个列的维度;z_data为x_data加了一个行的维度。