如何选择一列或其他列

Question

我希望达到的目标：

我希望（基于另一个数组）从numpy数组中选择一列或另一列

可复制的示例

我有以下数组

a = np.array([["a", "b"], 
              ["a", "b"], 
              ["a", "b"], 
              ["a", "b"]])

我希望使用以下条件选择：

select = [0, 1, 0, 1]

我希望得到：

np.array(["A",
          "B",
          "A",
          "B"])

NB

注1：我知道我可以做一个for循环，但它不是真的有效，所以我在寻找一个更好的答案。你知道吗

注2：我想这很简单，但我似乎没有合适的谷歌关键词：/

注3：这是一个简单的例子，在完美世界中，解决方案适用于任意数量的列和任意数量的维度（例如选择n轴）

谢谢

谢谢你的帮助/指点

Answer 1

你可以用这个我想：

result = np.array([a[idx][selectElem] for idx, selectElem in enumerate(select)])

这叫做列表理解。你知道吗