#!/usr/bin/env python
# coding: utf-8

# # 对角线

# 这里，使用与之前不同的导入方法：

# 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


# 查看它的对角线元素：

# In[3]:


a.diagonal()


# 可以使用偏移来查看它的次对角线，正数表示右移，负数表示左移：

# In[4]:


a.diagonal(offset=1)


# In[5]:


a.diagonal(offset=-1)


# 可以使用花式索引来得到对角线：

# In[6]:


i = [0,1,2]
a[i, i]


# 可以更新对角线的值：

# In[7]:


a[i, i] = 2
a


# 修改次对角线的值：

# In[8]:


i = np.array([0,1])
a[i, i + 1] = 1
a


# In[9]:


a[i + 1, i] = -1
a