用列表作为列表索引

a = np.around(10*np.random.random((3, 3)))
b = [0, 1, 2]
c = [0, 1, 2]
print(a)
print(a[b, c])

运行结果如下,利用[0, 1, 2]作为行索引, [0,1, 2]作为列索引,输出了a[0, 0], a[1, 1], a[2, 3]。

 

posted @ 2019-08-03 11:26  地球上最后一个直男  阅读(633)  评论(0编辑  收藏  举报