numpy 常用方法笔记

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
import numpy as np
a = np.ones((2,3))
"""
a:
array([[ 1., 1., 1.],
[ 1., 1., 1.]])
"""
b = np.array([[2,3,4],[5,6,7]])
a =a - b
"""
a:
array([[ 1., 2., 3.],
[ 4., 5., 6.]])
"""
np.random.shuffle(a[0])
"""
打乱顺序,可添加随机种子
return None
a:
array([[ 3., 1., 2.],
[ 4., 5., 6.]])
"""
np.argsort(a,axis = 0)
"""
返回从小到大的index
array([[0, 0, 0],
[1, 1, 1]])
"""
np.argsort(a,axis = 1)
"""
对a[0]而言,最小的元素是1,则返回的矩阵的第0行第0个元素为元素1的下标。
array([[1, 2, 0],
[0, 1, 2]])
"""
np.transpose(a)
"""
矩阵转置,一维矩阵需转化为二维
array([[ 3., 4.],
[ 1., 5.],
[ 2., 6.]])
"""
a = a.astype(int)
"""
a:
array([[3, 1, 2],
[4, 5, 6]])
"""
np.bincount(a.reshape(-1))
"""
返回数组中从0到最大值的每个整数数量列表,0没有,1没有,1一个
array([0, 0, 1, 1, 2, 2, 2, 2, 1, 1])
"""