矩阵坐标的指定

# -*- coding: utf-8 -*-
import numpy as np
data = np.array([[1, 2, 3],
                 [4, 5, 6],
                 [7, 8, 9]])
i = [0, 1, 2]
j = np.array([2, 1, 0])
print('data[i, j]:\n', data[i, j]) # 指定矩阵的坐标的时候，记住需要是numpy的array而不是python自己的列表格式