Python NumPy中的diag函數

NumPy包中的內置diag函數頗有意思。python 假設建立一個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],

>>阅读原文<<