Skip to content

Numpy 简介

导入numpy

Numpy是Python的一个很重要的第三方库,很多其他科学计算的第三方库都是以Numpy为基础建立的。

Numpy的一个重要特性是它的数组计算。

在使用Numpy之前,我们需要导入numpy包:

python
from numpy import *

使用前一定要先导入 Numpy 包,导入的方法有以下几种:

python
import numpy
    import numpy as np
    from numpy import *
    from numpy import array, sin

事实上,在ipython中可以使用magic命令来快速导入Numpy的内容。

python
%pylab

数组上的数学操作

假如我们想将列表中的每个元素增加1,但列表不支持这样的操作(报错)

转成 array

array 数组支持每个元素加 1 这样的操作(适用基本的运算法则)

python
a = array(a)
a + 1

提取数组中的元素

python
# 提取第一个元素:
a[0]

# 提取前两个元素:
a[:2]

# 最后两个元素:
a[-2:]

# 将它们相加:
a[:2] + a[-2:]

修改数组形状

查看 array 的形状:

python
a.shape

修改 array 的形状:

python
a.shape = 2,2

多维数组

a 现在变成了一个二维的数组,可以进行加法:

乘法仍然是对应元素的乘积,并不是按照矩阵乘法来计算:

python

画图

linspace 用来生成一组等间隔的数据:

python
a = linspace(0, 2*pi, 21)
%precision 3

三角函数:

python
b = sin(a)

画出图像:

python
%matplotlib inline
plot(a, b)

从数组中选择元素

假设我们想选取数组b中所有非负的部分,首先可以利用 b 产生一组布尔值:

python
b >= 0
mask = b >= 0

画出所有对应的非负值对应的点:

python
plot(a[mask], b[mask], 'ro')

© 2023-2024 LiuJingcheng. 保留所有权利。