这里,使用与之前不同的导入方法:
In [1]:
import numpy as np
使用numpy中的函数前,需要加上 np.
:
In [2]:
a = np.array([11,21,31,12,22,32,13,23,33])
a.shape = 3,3
a
Out[2]:
查看它的对角线元素:
In [3]:
a.diagonal()
Out[3]:
可以使用偏移来查看它的次对角线,正数表示右移,负数表示左移:
In [4]:
a.diagonal(offset=1)
Out[4]:
In [5]:
a.diagonal(offset=-1)
Out[5]:
可以使用花式索引来得到对角线:
In [6]:
i = [0,1,2]
a[i, i]
Out[6]:
可以更新对角线的值:
In [7]:
a[i, i] = 2
a
Out[7]:
修改次对角线的值:
In [8]:
i = np.array([0,1])
a[i, i + 1] = 1
a
Out[8]:
In [9]:
a[i + 1, i] = -1
a
Out[9]: