Numpy用法小結
阿新 • • 發佈:2019-01-12
匯入numpy庫:
import numpy as np
1、建立向量/矩陣:
arr = np.array([[1,2,3],[4,5,6]])
'''array([[1, 2, 3],
[4, 5, 6]])'''
arr.shape
(2, 3)
mat = np.matrix([[2,3,4],[5,6,7]])
'''matrix([[2, 3, 4],
[5, 6, 7]])'''
mat.shape
(2,3)
2.拼接矩陣:
import numpy as np
a = np.array([1, 2, 3])
b = np.array([4, 5, 6])
c = np.r_[a,b]
d = np.c_[a,b]
c
Out[11]: array([1, 2, 3, 4, 5, 6])
c.shape
Out[12]: (6,)
d
Out[13]:
array([[1, 4],
[2, 5],
[3, 6]])
d.shape
Out[14]: (3, 2)
np.r_是按列連線兩個矩陣,就是把兩矩陣上下相加,要求列數相等,類似於pandas中的concat()。
np.c_是按行連線兩個矩陣,就是把兩矩陣左右相加,要求行數相等,類似於pandas中的merge()。
3、
In [2]: a = np.array([[1,2,3],[4,5,6]])
In [3]: mask = np.random.rand(*a.shape) >= 0.9
In [4]: mask
Out[4]:
array([[ True, True, False],
[False, False, False]])
In [5]: a*mask
Out[5]:
array([[1, 2, 0],
[0, 0, 0]])
4、
In [2]: np.arange(10)
Out[2]: array([0, 1, 2, 3, 4, 5, 6 , 7, 8, 9])