首页 > Python教程 > Python NumPy库 阅读数:5

Python NumPy库的安装和使用

NumPy 就是一个数学运算的库,其是用 C 语言实现的,所以运算速度非常快。该模块也不是 Python 自带的,需要自行安装。

可以使用 PIP 进行安装,命令如下:

pip install numpy

使用该模块之前需要将其引入,常用方法是:

import numpy as np

这样以后就可以使用 np 来表示该模块了。

NumPy 最常见的数据结构是 ndarray,ndarray 表示 N-dimentioanl Array,就是多维数组的意思。本节也从这里开始介绍 NumPy。

ndarray的构造

可以使用多种方式来构建多维数组,最常见的是使用列表来构建多维数组。下面的例子便使用一维列表构建了一个一维数组。
>>> import numpy as np
>>> nda1 = np.array([1, 2, 3])        # 使用一维列表来作为输入
>>> nda1                       
array([1, 2, 3])6    >>> type(nda1)
<class 'numpy.ndarray'>

如果希望构建二维数组,可以使用下面的方法:
>>> input_list = [
...     [1, 2, 3],
...     [4, 5, 6]
... ]
>>> nda2 = np.array(input_list)
>>> nda2
array([[1, 2, 3],                    # 查看值
       [4, 5, 6]])
>>> type(nda2)                        # 查看类型
<class 'numpy.ndarray'>

也可以指定一些特征值,让 NumPy 自动产生相关的数组。例如指定维度,让其产生所有元素都为 0 的数组,代码如下:
>>> np.zeros(5)                        # 5个元素的一维数组
array([0., 0., 0., 0., 0.])
>>> np.zeros((5, 2))                # 二维数组,5行,2列
array([[0., 0.],
       [0., 0.],
       [0., 0.],
       [0., 0.],
       [0., 0.]])

也可以指定维度,让其产生所有元素值都为 1 的数组,代码如下:
>>> np.ones((5, 2))                    # 二维数组,5行,2列,所有元素都为1
array([[1., 1.],
       [1., 1.],
       [1., 1.],
       [1., 1.],
       [1., 1.]])
>>> np.ones(5)                        # 一维数组,5个元素
array([1., 1., 1., 1., 1.])

还可以让 NumPy 自动产生等差数组,此时需要指定开始值、结束值和步长。代码如下:
>>> np.arange(3,7,2)                # 从3开始,直到7,步长为2
array([3, 5])
>>> np.arange(3,7,1)                # 从3开始,直到7,步长为1
array([3, 4, 5, 6])
>>> np.arange(7, 3, -1)                # 从7开始,直到3,步长为-1
array([7, 6, 5, 4])
>>> np.arange(7, 3, -2)                # 从7开始,直到3,步长为-2
array([7, 5])

arange() 函数和 range() 类似,如果仅提供一个值,那么开始值是 0,步长是 1,代码如下:
>>> np.arange(7)
array([0, 1, 2, 3, 4, 5, 6])
如果提供两个参数,那么步长为 1:
>>> np.arange(2, 5)                # 从2开始,直到5,步长为1
array([2, 3, 4])
>>> np.arange(2, 6)                # 从2开始,直到6,步长为1
array([2, 3, 4, 5])

另外一个等差数列函数是 linspace(),其指定开始位置和结束位置,但不指定步长,而是指定元素个数。例如从 1 开始,到 5 结束,一共有 8 个数,那么生成的数组如下面所示:
>>> np.linspace(1, 5, 8)        # 包括1和5,等分8个点
array([1. , 1.57142857, 2.14285714, 2.71428571, 3.28571429,
     3.85714286, 4.42857143, 5])
可以发现元素个数和指定的一致,开始值和结束值也都被包含,而且它们的确是等差数列。

linspace() 函数比较有用,例如要画正弦函数在 0 到 2π 之间的图形,便可以使用该函数在 0 到 2π 之间产生均匀分布的 100 个点,然后使用 matplotlib 将它们画出来。下面是演示的代码:
import matplotlib.pyplot as plt
import numpy as np
x  = np.linspace(0, 2*np.pi, 100)
y  = [np.sin(e) for e in x]
plt.plot(x, y)
plt.savefig("sindemo1.png")
运行后产生的图片如图 1 所示。


图 1 y=sin(x)的曲线

还可以使用 logspace() 函数让 NumPy 自动产生等比数列,此时需要指定开始点和结束点,同时指定点的个数。如果没有提供点的数目,默认是生成 50 个点。
>>> np.logspace(2.0, 3.0, num=4)    # 4个点,其实位置是102,结束位置是103
array([ 100. , 215.443469,  464.15888336, 1000.])

下面是一个例子,其演示了 logspace() 的用法和参数 endpoint 的用法。endpoint=True 表示结束值被包含在输出数组中,否则表示不包含在输出数组中。下面是完整的代码:
import matplotlib.pyplot as plt
import numpy as np
N = 10            # 一共10个点
x1 = np.logspace(0.1, 1, N, endpoint=True)        # 10被算作是最后一个点
x2 = np.logspace(0.1, 1, N, endpoint=False)    # 10不被算作是最后一个点
y = np.zeros(N)
plt.plot(x1, y, 'o')
plt.plot(x2, y + 0.5, 'x')
plt.ylim([-0.5, 1])                                # y轴的范围是-0.5到1
plt.savefig("logspace1.png")                    # 保存图片到文件
运行该脚本,得到的输出图片如图 2 所示。


图 2 logspace()的用法

还可以使用 full() 函数指定维度和一个值,让所有的元素都等于该值。该函数和 ones() 类似,但值是由用户指定的。
>>> np.full((2, 2), np.inf)                # 所有元素都是无穷大
array([[inf, inf],
       [inf, inf]])
>>> np.full((2, 2), 11)                    # 所有元素都是11
array([[11, 11],
       [11, 11]])
>>> np.full((2, 2), 1.51)                # 所有元素都是1.51
array([[1.51, 1.51],
       [1.51, 1.51]])

使用 eye() 函数还可以自动生成单位矩阵,就是仅对角线上的值为 1,其他位置上的值都为 0。
>>> np.eye(2)                                    # 2x2的单位矩阵
array([[1., 0.],
       [0., 1.]])
>>> np.eye(3)                                    # 3x3的单位矩阵
array([[1., 0., 0.],
       [0., 1., 0.],
       [0., 0., 1.]])

还可以自动产生随机的矩阵,例如可以使用 random.normal() 函数产生一个正态分布的一维矩阵:
>>> mu, sigma = 0, 0.1                    # mu是平均值,sigma代表分散程度
>>> s = np.random.normal(mu, sigma, 1000)
>>> s.size                                        # 元素个数为1000
1000
>>> np.mean(s)                                    # 平均值接近0
-0.0011152161285000821
>>> abs(mu - np.mean(s)) < 0.01                # 平均值接近mu=0
True
>>> abs(sigma - np.std(s, ddof=1)) < 0.01        # 分散程度检查
True

可以将生成的数据画出来,使用下面的代码:
import matplotlib.pyplot as plt
import numpy as np
mu, sigma = 0, 0.1
s = np.random.normal(mu, sigma, 1000)
count, bins, ignored = plt.hist(s, 30, density=True)
plt.plot(bins, 1/(sigma * np.sqrt(2 * np.pi)) *
         np.exp( - (bins - mu)**2 / (2 * sigma**2) ),
         linewidth=2, color='r')
plt.savefig("rand1.png")
运行该脚本,得到的输出图片文件如图 3 所示。


图 3 正态分布

还可以生成完全随机的矩阵,方法是使用 np.random.rand(外形)函数。例如在下面的例子中,就生成了随机内容组成的指定外形的矩阵。
>>> np.random.rand(3,2)                    # 3x2的二维矩阵
array([[0.11319256, 0.84668147],
       [0.4040353 , 0.70912343],
       [0.6511614 , 0.80706271]])
>>> np.random.rand(3,2,2)                # 3x2x2的三维矩阵
array([[[0.64851863, 0.3895985 ],
        [0.63038544, 0.58402249]],
       [[0.39816687, 0.92149102],
        [0.07113285, 0.17109903]],
       [[0.06713956, 0.39415293],
        [0.06125844, 0.71276929]]])
>>> np.random.rand(4)                    # 一维矩阵
array([0.11918788, 0.91847982, 0.29599804, 0.42242323])