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],

>>阅读原文<<