numpy - 获取嵌套数组中的元素

Question

我有一个np.array= [[1,2,3,4],[2,1,4,1],[0,9,2,1],...] 我可以在没有循环的情况下以某种方式获取每个嵌套数组中的每个第二个元素，我想得到[[2],[1],[9]] 或者这只能通过循环来完成？

标签： numpy

Answer 1

如果你想获得每一行的第二个元素

arr = np.array([[1,2,3,4],[2,1,4,1],[0,9,2,1]]) 
arr[:, 1]