如何在 python 中将 numpy 数组拆分为单独的数组。分割数由用户指定，分割基于索引

Question

我想将我的 numpy 数组拆分为单独的数组。分离必须基于索引。分割计数由用户给出。

例如， 输入数组：my_array=[1,2,3,4,5,6,7,8,9,10]

如果用户给出分割计数=2， 那么，分割必须类似于

my_array1=[1,3,5,7,9]
my_array2=[2,4,6,8,10]

如果用户给出分割计数=3，那么 输出数组必须是

my_array1=[1,4,7,10]
my_array2=[2,5,8]
my_array3=[3,6,9]

任何人都可以解释一下，我使用偶奇概念对拆分计数 2 进行了操作，

for i in range(len(outputarray)):
    if i%2==0:
        even_array.append(outputarray[i])  
    else:
        odd_array.append(outputarray[i])

我不知道如何根据索引对变量计数（如 3、4、5）进行拆分。

Answer 1

您可以使用矢量索引（又名 花式索引) 对于它：

>>> a=np.array([1,2,3,4,5,6,7,8,9,10])
>>> n = 3
>>> [a[np.arange(i, len(a), n)] for i in range(n)]

[array([ 1,  4,  7, 10]), array([2, 5, 8]), array([3, 6, 9])]

解释

arange(i, len(a), n) 生成一个以以下开头的整数数组i，跨度不超过 len(a)，步长为 n。例如，对于 i = 0 它会生成一个数组

 >>> np.arange(0, 10, 3)
 array([0, 3, 6, 9])

现在，当您使用另一个数组对一个数组进行索引时，您将获得所请求索引处的元素：

 >>> a[[0, 3, 6, 9]]
 array([ 1,  4,  7, 10])

对于 i=1..2 重复这些步骤，结果是所需的数组列表。

Answer 2

这是执行任务的仅 Python 方法

def split_array(array, n=3):
    arrays = [[] for _ in range(n)]
    for x in range(n):
        for i in range(n):
            arrays[i] = [array[x] for x in range(len(array)) if x%n==i]
    return arrays

输入：

my_array=[1,2,3,4,5,6,7,8,9,10]
print(split_array(my_array, n=3))

输出：

[[1, 4, 7, 10], [2, 5, 8], [3, 6, 9]]