python - 在这个 numpy 数组与不同数组相乘的 numpy 问题中发生了什么？

Question

import numpy as np

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

y = x [[0,1,2], [0,1,0]]   #:i did not understand this step,what is happening here?

print y

OUTPUT: [1 4 5]

标签： pythonnumpynumpy-ndarray

Answer 1

当你在做

x[a, b]

a 和 b 是数组，您指定了一系列要使用的索引。例如，在这里你说“选择第 0 行，然后是第 1 行，然后是第 2 行”和“选择第 0 列，第 1 和第 0 行”。

所以，你得到的数组是 [x[0,0], x[1,1], x[2,0]]