2016-01-11 55 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]

第一個索引列表包含第一個軸(矩陣表示法:行索引)的索引,第二個列表包含第二個軸(列索引)的索引。