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')