Bagaimana cara mengindeksnya? Mengapa keluaran [1,4,5]?

Saya mengikuti tutorial di http://cs231n.github.io/python-numpy -tutorial/#numpy

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]"
1
Ab Be 20 September 2019, 08:23

1 menjawab

Jawaban Terbaik

Ini disebut pengindeksan mewah di numpy.

Anda dapat membayangkan daftar pertama dan daftar kedua sebagai sumbu x dan sumbu y. Jadi a[[0,1,2],[0,1,0]] seperti mendapatkan tiga elemen yang koordinatnya (0,0), (1,1), (2,0) dari a.

a[0,0] # 1
a[1,1] # 4
a[2,0] # 5
3
zihaozhihao 20 September 2019, 05:49