如何通过一一跳过每一列来创建复杂的numpy数组答案

【问题标题】：How to create complex numpy array by skipping each column one by one如何通过一一跳过每一列来创建复杂的numpy数组
【发布时间】：2021-07-27 23:26:48
【问题描述】：

我有一个像这样的复杂 (3x6) numpy 数组：

x = [[(1+j2) ,(2+j8), (4+j1), (6+j8), (7+j3), (8+j2)],
     [(3+j8), (5+j1), (7+j5), (3+j2), (6+j1), (3+j1)],
     [(1+j5), (5+j4), (2+j9), (9+j5), (8+j1), (4+j1)]]

我想将它们拆分为：

x1 = [[(1+j2), (4+j1), (7+j3)],
      [(3+j8), (7+j5), (6+j1)],
      [(1+j5), (2+j9), (8+j1)]]

和

x2 = [[(2+j8), (6+j8), (8+j2)],
      [(5+j1), (3+j2), (3+j1)),
      [(5+j4), (9+j5), (4+j1)]]

我的意思是，我想通过一一跳过每一列来获得一个新的复杂 numpy 数组。我该怎么做？

【问题讨论】：

标签： python arrays python-3.x numpy numpy-ndarray

【解决方案1】：

这可以使用数组索引来实现。

import numpy as np

x = np.array(
    [
        [(1 + 2j), (2 + 8j), (4 + 1j), (6 + 8j), (7 + 3j), (8 + 2j)],
        [(3 + 8j), (5 + 1j), (7 + 5j), (3 + 2j), (6 + 1j), (3 + 1j)],
        [(1 + 5j), (5 + 4j), (2 + 9j), (9 + 5j), (8 + 1j), (4 + 1j)],
    ]
)
x1 = x[:, ::2]
x2 = x[:, 1::2]

可以找到数组索引更详细的解释here

【讨论】：

非常感谢。

【解决方案2】：

为此，您可以使用list comprehension

示例：

x = [[(1+j2) ,(2+j8), (4+j1), (6+j8), (7+j3), (8+j2)],
     [(3+j8), (5+j1), (7+j5), (3+j2), (6+j1), (3+j1)],
     [(1+j5), (5+j4), (2+j9), (9+j5), (8+j1), (4+j1)]]

x1 = [ i[0:6:2] for i in x ]
x2 = [ i[1:6:2] for i in x ]

【讨论】：