Python NumPy中的diag函式

摘自：

numpy包中的內建diag函式很有意思。

假設建立乙個1維陣列a，和乙個3*3陣列b：

import numpy as np

a = np.arange(1, 4)
b = np.arange(1, 10).reshape(3, 3)

>>> a

array([1, 2, 3])
>>> b
array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
使用diag函式，看

>>> np.diag(a)

array([[1, 0, 0],
[0, 2, 0],
[0, 0, 3]])
>>> np.diag(b)
array([1, 5, 9])

array是乙個1維陣列時，結果形成乙個以一維陣列為對角線元素的矩陣

array是乙個二維矩陣時，結果輸出矩陣的對角線元素

python numpy中nonzero 的用法

當使用布林陣列直接作為下標物件或者元組下標物件中有布林陣列時，都相當於用nonzero 將布林陣列轉換成一組整數陣列，然後使用整數陣列進行下標運算。nonzeros a 返回陣列a中值不為零的元素的下標，它的返回值是乙個長度為a.ndim 陣列a的軸數 的元組，元組的每個元素都是乙個整數陣列，其值為...

python numpy中nonzero 的用法

當使用布林陣列直接作為下標物件或者元組下標物件中有布林陣列時，都相當於用nonzero 將布林陣列轉換成一組整數陣列，然後使用整數陣列進行下標運算。nonzeros a 返回陣列a中值不為零的元素的下標，它的返回值是乙個長度為a.ndim 陣列a的軸數 的元組，元組的每個元素都是乙個整數陣列，其值為...

python numpy中nonzero 的用法

當使用布林陣列直接作為下標物件或者元組下標物件中有布林陣列時，都相當於用nonzero 將布林陣列轉換成一組整數陣列，然後使用整數陣列進行下標運算。nonzeros a 返回陣列a中值不為零的元素的下標，它的返回值是乙個長度為a.ndim 陣列a的軸數 的元組，元組的每個元素都是乙個整數陣列，其值為...