1. 程式人生 > >Numpy用法小結

Numpy用法小結

匯入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])