python 科学计算 — numpy 学习笔记

study / python learning
字数统计: 2.1k   |   阅读时长≈ 10 分钟
阅读数:0次

python numpy的学习笔记 要使用numpy，首先需要import进来：

建议使用别名np，因为你会发现这几乎是一个大家都在用的写法！

1
import numpy as np

 

一、创建数组

1.和list类似，可以直接用arange来表示范围：

1
2
3
import numpy as np
a = np.arange(10)
print a

Output:

1
[0 1 2 3 4 5 6 7 8 9]

 

从[2,20)间隔为3的等差数列

1
2
3
import numpy as np
a = np.arange(2,20,3)
print a

Output:

1
2
[ 2  5  8 11 14 17]

 

类似arange，linspace从[start , stop ] 生成num个数，Num个数间隔相等。（默认为float）

linspace(start, stop, num=50, endpoint=True, retstep=False, dtype=None)

1
print np.linspace(0, 2, 9)

Output:

1
[ 0.    0.25  0.5   0.75  1.    1.25  1.5   1.75  2.  ]

 

2.用list/tuple创建数组(list也可以是二维的)

1
2
3
import numpy as np
a = np.array([range(5)])
print a

 

3.快速生成x*y的全零数组

1
2
d=(2,3) #(2,3,4)变为3维
print np.zeros(d)

Output:

1
2
[[ 0.  0.  0.]
 [ 0.  0.  0.]]

 

默认生成的类型是浮点型 可以用dtype改为int

1
2
d = (4, 5)
print np.ones(d, dtype=int)

还可以通过zeros_like生成相同形状的全0数组，如：

1
img = np.zeros_like(image)

 

0~1的随机数：

1
np.random.rand(5)

 

3.一维转化为二维数组 reshape

1
2
3
a= np.arange(10)
a = a.reshape(2, 5)
print a

 

Output:

1
2
[[0 1 2 3 4]
 [5 6 7 8 9]]

PS:

这个会根据列来自动适应行！

下面这句在a=np.arange(10)等价于a = a.reshape(2, 5)

在a=np.arange(20)等价于a = a.reshape(4, 5)！

1
a.reshape(-1, 5)

 

甚至可以转化为更高维的：

1
2
3
4
import numpy as np
a = np.arange(20)
a = a.reshape(2,2,5)
print a

Output:

1
2
3
4
5
6
7
[[[ 0  1  2  3  4]
  [ 5  6  7  8  9]]

 [[10 11 12 13 14]
  [15 16 17 18 19]]]


 

当然，也可以二维转化一维的：ravel

1
2
3
import numpy as np
a = np.array([[1,2,3],[4,5,6]])
print np.ravel(a)

Output:

1
[1 2 3 4 5 6]

 

4. 切割vsplit和hsplit

vsplit和hsplit用法差不多，差别在于：vsplit来进行分行，而hsplit来分列（即作用于同一行中）。

numpy.vsplit(ary, indices_or_sections)

（1）指定切分个数（需要行数能被切分的个数整除）

看下面的代码，生成了6*3的数组，用vsplit切分为3个。

1
2
3
4
import numpy as np
a = np.arange(18).reshape(-1,3)
print a,'\n'
print np.vsplit(a,3)

Output:

1
2
3
4
5
6
7
8
9
10
11
12
13
[[ 0  1  2]
 [ 3  4  5]
 [ 6  7  8]
 [ 9 10 11]
 [12 13 14]
 [15 16 17]] 

[array([[0, 1, 2],
       [3, 4, 5]]), array([[ 6,  7,  8],
       [ 9, 10, 11]]), array([[12, 13, 14],
       [15, 16, 17]])]


 

（2）指定位置切分

我只改动了最后一行，将vsplit第二个参数改为一个List，来指定切分的位置。

1
2
3
4
import numpy as np
a = np.arange(18).reshape(-1,3)
print a,'\n'
print np.vsplit(a,[1,5])

Output:

1
2
3
4
5
6
7
8
9
10
11
12
13
[[ 0  1  2]
 [ 3  4  5]
 [ 6  7  8]
 [ 9 10 11]
 [12 13 14]
 [15 16 17]] 

[array([[0, 1, 2]]), array([[ 3,  4,  5],
       [ 6,  7,  8],
       [ 9, 10, 11],
       [12, 13, 14]]), array([[15, 16, 17]])]


 

5.查看数组属性

ndim :维度 shape:各维度的大小 size:全部的元素个数 dtype:查看元素类型 type:数组的类型

1
2
3
4
5
6
7
8
import numpy as np
a = np.arange(20)
a = a.reshape(2,2,5)
print a.ndim    #a的维度
print a.shape   #a的各维度的大小
print a.size    #a的全部的元素个数
print a.dtype   #a的元素类型
print type(a)   #a的类型

Output:

1
2
3
4
5
3
(2L, 2L, 5L)
20
int32
<type 'numpy.ndarray'>

 

二、数组操作

1.四则运算

+'，'-'，'*'，'/'运算都是基于全部的数组元素的

1
2
3
4
import numpy as np
a = np.array([1,2])
b= np.array([3,4])
print a+b

Output:

1
[4 6]

 

2.开根号/指数

1
2
3
4
5
6
7
import numpy as np
a = np.array([1,2])
print np.sqrt(a)    
print np.exp(a)     
print np.square(a)
print np.power(a,5)
print a ** 5        #可以用Power也可以直接 ** ,当然这么写比较爽

Output:

1
2
3
4
5
6
[ 1.          1.41421356]
[ 2.71828183  7.3890561 ]
[1 4]
[ 1 32]
[ 1 32]

 

3.数组最大最小值

1
2
3
4
5
6
7
8
import numpy as np
a = np.arange(20).reshape(4,5)
print a
print a.min()
print a.max()
print a.sum()
print a.min(axis=0) #minimun element in each column
print a.min(axis=1) #minimun element in each row

Output:

1
2
3
4
5
6
7
8
9
10
[[ 0  1  2  3  4]
 [ 5  6  7  8  9]
 [10 11 12 13 14]
 [15 16 17 18 19]]
0
19
190
[0 1 2 3 4]
[ 0  5 10 15]

numpy.ptp(a, axis=None, out=None)

返回最大和最小值之差

4.数组的均值/中位数

1
2
3
4
import numpy
a =range(5) +[1000]
print a
print numpy.mean(a),numpy.median(a)

Output:

1
2
[0, 1, 2, 3, 4, 1000]
168.333333333 2.5

 

5.数组取值

可以直接用下标对齐取值，注意直接赋值的话是浅拷贝！（即a赋给b，意味着b指向了a对应数据的内存地址）

想要真正的复制一份a给b，可以使用copy：

1
2
3
4
5
6
7
8
# -*- coding:utf-8 -*-
import numpy as np
a = np.array([[1, 2], [3, 4]])
b = a           #浅拷贝！
c = a.copy()  #深拷贝 np.copy(a)
print a[1,1],a[1][1] 
b[1][1]=5
print a[1][1],b[1][1],c[1][1]

Output:

1
2
4 4
5 5 4

 

利用':'可以访问到某一维的全部数据:

1
2
3
4
5
6
7
# -*- coding:utf-8 -*-
import numpy as np
a = np.arange(20).reshape(4, 5)
print a
print '\n',a[:,[1,3]] #取出a的每一行的第2个和第4个元素
print '\n',a[:,2][a[:,0] > 5] #取出a中的第3列的元素，如果该行第1个元素 ＞５

Output:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
[[ 0  1  2  3  4]
 [ 5  6  7  8  9]
 [10 11 12 13 14]
 [15 16 17 18 19]]

[[ 1  3]
 [ 6  8]
 [11 13]
 [16 18]]

[12 17]



 

6.数组拼接

使用hstack 横向拼接

使用vstack 纵向拼接

1
2
3
4
5
6
7
8
import numpy as np
import numpy.linalg as nlg
a = np.array([1,2,3])
b = np.array([4,5,6])
c = np.hstack([a,b])
d = np.vstack([a,b])
print c
print d

Output:

1
2
3
[1 2 3 4 5 6]
[[1 2 3]
 [4 5 6]]

 

三、矩阵对象

矩阵对象和数组的主要有两点差别：

  • 矩阵是二维的，而数组的可以是任意正整数维
  • 矩阵的''操作符进行的是矩阵乘法，乘号左侧的矩阵列和乘号右侧的矩阵行要相等，而在数组中''操作符进行的是每一元素的对应相乘，乘号两侧的数组每一维大小需要一致

1.创建矩阵

1
2
3
4
5
6
7
8
9
10
# -*- coding:utf-8 -*-
import numpy as np
a = np.arange(5)
a = np.mat(a)  #use np.mat(a) or np.asmatrix(a)
print a,type(a)

b = np.mat('1.0 2.0; 3.0 4.0') # you can use np.matrix()
print b,type(b)


Output:

1
2
3
4
[[0 1 2 3 4]] <class 'numpy.matrixlib.defmatrix.matrix'>
[[ 1.  2.]
 [ 3.  4.]] <class 'numpy.matrixlib.defmatrix.matrix'>

 

2.矩阵乘法

为了比较和数组的乘法，特地将b设为单位阵。

1
2
3
4
5
6
7
8
9
10
11
# -*- coding:utf-8 -*-
import numpy as np
a = np.array([[1, 2], [3, 4]])
b = np.array([[1, 0], [0, 1]])
print 'a=', a, '\nb=', b
print
print 'array : a*b:'
print a * b
print 'matrix: a*b:'
print np.asmatrix(a) * np.asmatrix(b)

Output:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
a= [[1 2]
 [3 4]] 
b= [[1 0]
 [0 1]]

array : a*b:
[[1 0]
 [0 4]]
matrix: a*b:
[[1 2]
 [3 4]]



 

3.矩阵转置

1
2
3
4
5
# -*- coding:utf-8 -*-
import numpy as np
a = np.array([[1, 2, 3], [4, 5, 6]])
print np.transpose(a)   #对于数组，用transpose
print np.matrix(a).T    #对于矩阵，直接T

Output:

1
2
3
4
5
6
[[1 4]
 [2 5]
 [3 6]]
[[1 4]
 [2 5]
 [3 6]]

 

4.矩阵求逆

1
2
3
4
5
6
import numpy as np
import numpy.linalg as nlg
a = np.mat([[1,2],[4,4]]) 
print a
b = nlg.inv(a)
print a*b

Output:

1
2
3
4
[[1 2]
 [4 4]]
[[ 1.  0.]
 [ 0.  1.]]

 

5.特征值和特征向量

1
2
3
4
5
6
7
8
import numpy as np
import numpy.linalg as nlg
a = np.random.rand(3,3)
eig_value, eig_vector = nlg.eig(a)
print "eigen value:"
print eig_value
print "eigen vector:"
print eig_vector

Output:

1
2
3
4
5
6
7
eigen value:
[ 1.35205837 -0.22652725  0.12474347]
eigen vector:
[[-0.66475046 -0.42263921  0.031526  ]
 [-0.54161542  0.70097608 -0.45506146]
 [-0.51454792 -0.57446378  0.88990178]]

 

四、其它

1.缺失值处理

NumPy用nan作为缺失值，可以用isnan判定：

1
2
3
4
import numpy as np
a = np.random.rand(2,2)
a[0, 1] = np.nan
print np.isnan(a)

nan_to_num()可用来将nan替换成0

 2.统计函数

np.bincount(a)：对整数数组中各个元素出现的次数进行统计，它要求数组中所有元素都是非负的，返回值中，第i个元素表示第i个元素在a中出现的次数。

 3.比较函数

直接用==比较

  • 对于相同长度的，返回一个numpy数组，每个元素为相应位置上是否相等.
  • 长度不相同，返回False
1
2
3
4
5
import numpy as np
a = np.arange(5)
b = np.array([0,1,2,4,3])
print a==b
print a==np.array([0,1,2,4,3,5])

Output:

1
2
[ True  True  True False False]
False

 

参考资料：

打赏
  • 版权声明： 本博客所有文章除特别声明外，著作权归作者所有。转载请注明出处！
  • Copyrights © 2013-2021 hrwhisper

请我喝杯咖啡吧~

支付宝
微信