2016-01-11 57 views
1

SOF,使用List的NumPy索引?

我注意到这个URL一个有趣的NumPy的演示:

http://cs231n.github.io/python-numpy-tutorial/

我看到这一点:

import numpy as np 

a = np.array([[1,2], [3, 4], [5, 6]]) 

# An example of integer array indexing. 
# The returned array will have shape (3,) and 

print(a[[0, 1, 2], [0, 1, 0]]) 
# Prints "[1 4 5]" 

我的理解使用整数作为指标参数:

a[1,1] 

和th是语法:

a[0:2,:] 

一般来说, 如果我使用列表作为指数的语法,这是什么意思?

具体来说, 我不明白为什么:

print(a[[0, 1, 2], [0, 1, 0]]) 
# Prints "[1 4 5]" 

回答

2

最后一条语句将打印(以矩阵形式)a(0,0)a(1,1)a(2,0)。在蟒蛇符号,这是a[0][0],a[1][1]a[2][0]

第一个索引列表包含第一个轴(矩阵表示法:行索引)的索引,第二个列表包含第二个轴(列索引)的索引。