生成数组的函数
arange
arange
类似于Python中的 range
函数,只不过返回的不是列表,而是数组:
arange(start, stop=None, step=1, dtype=None)
产生一个在区间 [start, stop)
之间,以 step
为间隔的数组,如果只输入一个参数,则默认从 0
开始,并以这个值为结束:
与 range
不同, arange
允许非整数值输入,产生一个非整型的数组:
数组的类型默认由参数 start, stop, step
来确定,也可以指定:
由于存在精度问题,使用浮点数可能出现问题:
stop
的值 2.1
出现在了数组中,所以使用浮点数的时候需要注意。
linspace
linspace(start, stop, N)
产生 N
个等距分布在 [start, stop]
间的元素组成的数组,包括 start, stop
。
logspace
logspace(start, stop, N)
产生 N 个对数等距分布的数组,默认以10为底:
产生的值为。
meshgrid
有时候需要在二维平面中生成一个网格,这时候可以使用 meshgrid
来完成这样的工作:
x
对应网格的第一维,y
对应网格的第二维。
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
from matplotlib import cm
def f(x, y):
# sinc 函数
r = np.sqrt(x ** 2 + y ** 2)
result = np.sin(r) / r
result[r == 0] = 1.0
return result
x_ticks = np.linspace(-10, 10, 51)
y_ticks = np.linspace(-10, 10, 51)
x, y = np.meshgrid(x_ticks, y_ticks)
z = f(x, y)
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.plot_surface(x, y, z,
rstride=1, cstride=1,
cmap=cm.YlGnBu_r)
ax.set_xlabel('x')
ax.set_ylabel('y')
ax.set_zlabel('z')
meshgrid
可以设置轴排列的先后顺序:
- 默认为
indexing='xy'
即笛卡尔坐标,对于2维数组,返回行向量x
和列向量y
- 或者使用
indexing='ij'
即矩阵坐标,对于2维数组,返回列向量x
和行向量y
。
ogrid , mgrid
Matlab中有 meshgrid
的用法:
meshgrid(-1:.5:1, -1:.5:1)
Numpy的 meshgrid
并不支持这样的用法,但我们可以使用 ogrid / mgrid
来实现类似这样的用法。
ogrid
与 mgrid
的区别在于:
ogrid
相当于meshgrid(indexing='ij', sparse=True)
mgrid
相当于meshgrid(indexing='ij', sparse=False)
注意:
- 这里使用的是中括号
- Matlab 使用的是
start:step:end
的表示,Numpy 使用的是start:end:step
的表示 - 这里的结果不包括
end
的值
为了包含 end
的值,我们可以使用这样的技巧:
我们在 step
的位置传入一个复数 5j
,表示我们需要一个 5
个值的数组,此时返回值就会包含 end
的值。
重复之前的画图:
y, x = np.ogrid[-10:10:51j, -10:10:51j]
z = f(x, y)
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.plot_surface(x, y, z,
rstride=1, cstride=1,
cmap=cm.YlGnBu_r)
ax.set_xlabel('x')
ax.set_ylabel('y')
ax.set_zlabel('z')
这里,我们交换了 x, y
输出值的顺序。
r_
, c_
我们可以使用 r_ / c_
来产生行向量或者列向量。
使用切片产生:
复数步长制定数组长度:
连接多个序列,产生数组:
列向量:
ones , zeros
ones(shape, dtype=float64) zeros(shape, dtype=float64)
产生一个制定形状的全 0
或全 1
的数组,还可以制定数组类型:
empty
empty(shape, dtype=float64, order='C')
也可以使用 empty
方法产生一个制定大小的数组(数组所指向的内存未被初始化,所以值随机),再用 fill
方法填充:
empty_
like, ones_
like, zeros_
like
empty_like(a)
ones_like(a)
zeros_like(a)
产生一个跟 a
大小一样,类型一样的对应数组。
identity
indentity(n, dtype=float64)
产生一个 n
乘 n
的单位矩阵: