如何使用切片在多个数组中获取 2 个数字（numpy）

Question

如果我有一个数组

a = np.array([[1,2,3],[4,5,6],[7,8,9]])

输出将是 =[1,2,3],[4,5,6],[7,8,9]

使用切片 [start:endindex:stepindex]， 我如何检索 3 和 7？

有可能吗？

我试过了

a[:3:2]

这给了我第一行和第三行

Answer 1

In [928]: a = np.array([[1,2,3],[4,5,6],[7,8,9]])                                                      
In [929]: a                                                                                            
Out[929]: 
array([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]])

[3,7] 在这个二维数组中不是常规模式。但它的平面视图：

In [931]: a.ravel()                                                                                    
Out[931]: array([1, 2, 3, 4, 5, 6, 7, 8, 9])
In [932]: a.ravel()[2::4]                                                                              
Out[932]: array([3, 7])
In [933]: a.flat[2::4]                                                                                 
Out[933]: array([3, 7])

现在保证它可以扩展到更大的数组和选择。