Numpy 按数组切片？

Question

import numpy as np
a = np.array([4, 3, 2, 1])
b = np.array([1, 2, 4, 3])
c = np.stack((a, b))
i = np.array([0, 1])
print(c[i])

我有：

[[4 3 2 1]
 [1 2 4 3]]

但我的预期输出：

[4 2]

我该如何实施？

Answer 1

你想在这里实现什么？

c = [[4 3 2 1] [1 2 4 3]]; 从 np.stack 调用

[c[0][0],c[1][1]]返回[4 2]这就是你想要的。