1. 程式人生 > >numpy 模組學習記錄三

numpy 模組學習記錄三

>>> b = np.eye(6)
>>> b
array([[1., 0., 0., 0., 0., 0.],
       [0., 1., 0., 0., 0., 0.],
       [0., 0., 1., 0., 0., 0.],
       [0., 0., 0., 1., 0., 0.],
       [0., 0., 0., 0., 1., 0.],
       [0., 0., 0., 0., 0., 1.]])
>>> c = b == 1
>>> c
array([[ True, False, False, False, False, False],
       [False,  True, False, False, False, False],
       [False, False,  True, False, False, False],
       [False, False, False,  True, False, False],
       [False, False, False, False,  True, False],
       [False, False, False, False, False,  True]])
>>> b[c]
array([1., 1., 1., 1., 1., 1.])
>>> b
array([[1., 0., 0., 0., 0., 0.],
       [0., 1., 0., 0., 0., 0.],
       [0., 0., 1., 0., 0., 0.],
       [0., 0., 0., 1., 0., 0.],
       [0., 0., 0., 0., 1., 0.],
       [0., 0., 0., 0., 0., 1.]])
>>> b[:4]
array([[1., 0., 0., 0., 0., 0.],
       [0., 1., 0., 0., 0., 0.],
       [0., 0., 1., 0., 0., 0.],
       [0., 0., 0., 1., 0., 0.]])
>>> c = b[:4]
>>> c
array([[1., 0., 0., 0., 0., 0.],
       [0., 1., 0., 0., 0., 0.],
       [0., 0., 1., 0., 0., 0.],
       [0., 0., 0., 1., 0., 0.]])
>>> c[c==0]
array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,
       0., 0., 0.])
>>> c[c==0] = 12
>>> c
array([[ 1., 12., 12., 12., 12., 12.],
       [12.,  1., 12., 12., 12., 12.],
       [12., 12.,  1., 12., 12., 12.],
       [12., 12., 12.,  1., 12., 12.]])
>>> c
array([[ 1., 12., 12., 12., 12., 12.],
       [12.,  1., 12., 12., 12., 12.],
       [12., 12.,  1., 12., 12., 12.],
       [12., 12., 12.,  1., 12., 12.]])
>>> f = np.array([True,False,True,False])
>>> f
array([ True, False,  True, False])
>>> c[f]
array([[ 1., 12., 12., 12., 12., 12.],
       [12., 12.,  1., 12., 12., 12.]])
>>> f = np.array([True,True,True,False,False,False])
>>> f
array([ True,  True,  True, False, False, False])
>>> c[:,f]
array([[ 1., 12., 12.],
       [12.,  1., 12.],
       [12., 12.,  1.],
       [12., 12., 12.]])
>>> c[1:3,f]
array([[12.,  1., 12.],
       [12., 12.,  1.]])
>>> c
array([[ 1., 12., 12., 12., 12., 12.],
       [12.,  1., 12., 12., 12., 12.],
       [12., 12.,  1., 12., 12., 12.],
       [12., 12., 12.,  1., 12., 12.]])
>>> c[:,f] = 0
>>> c
array([[ 0.,  0.,  0., 12., 12., 12.],
       [ 0.,  0.,  0., 12., 12., 12.],
       [ 0.,  0.,  0., 12., 12., 12.],
       [ 0.,  0.,  0.,  1., 12., 12.]])
>>> c[c==0] = 1
>>> c
array([[ 1.,  1.,  1., 12., 12., 12.],
       [ 1.,  1.,  1., 12., 12., 12.],
       [ 1.,  1.,  1., 12., 12., 12.],
       [ 1.,  1.,  1.,  1., 12., 12.]])