从阵列形成一对非连续元素

Question

生成器表达式有多个示例,从本网站的列表中产生成对的连续元素.但是,我想做一些不同的事情:

给定np.arange(6),我想输出[[0, 3], [1, 4], [2, 5]](这里,以列表的形式,但两者list和np.ndarray输出格式都可以).

当然,我可以通过for循环和快速和脏的numpy索引来做到这一点,但是有更优雅(并且希望更快)的方式吗？谢谢!

Answer 1

您可以reshape在指定Fortran顺序时使用(第一个索引更改最快):

In [104]: np.arange(6).reshape(3, -1, order='F')
Out[104]: 
array([[0, 3],
       [1, 4],
       [2, 5]])