分类：diag函数

2021
07-04

Python NumPy中diag函数的使用说明

NumPy包中的内置diag函数很有意思。假设创建一个1维数组a，和一个3*3数组b：importnumpyasnpa=np.arange(1,4)b=np.arange(1,10).reshape(3,3)结果如下：>>>aarray([1,2,3])>>>barray([[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])可以发现，当np.diag(array)中array是一个1维数组时，结果形成一个以一维数组为... 继续阅读 >