【问题标题】:Reconstruct numpy array using indexing [duplicate]使用索引重建numpy数组[重复]
【发布时间】:2018-06-24 19:16:11
【问题描述】:

我有一个由3*k 元素组成的numpy 数组,其中k 是一个整数。例如,对于k=3,我有数组A。下面,xyz 由来自 A 的绘图元素填充,如下例所示:

import numpy as np

A = np.arange(9)
>> array([0, 1, 2, 3, 4, 5, 6, 7, 8])

x = A[0::3]
>> array([0, 3, 6])

y = A[1::3]
array([1, 4, 7])

z = A[2::3]
>> array([2, 5, 8])

现在,我想使用xyz来重构A。有没有不使用for 循环的便捷方法?

【问题讨论】:

    标签: python numpy indexing


    【解决方案1】:

    你可以的

    np.array([x,y,z]).T.reshape(-1)
    

    【讨论】:

    • 它就像一个魅力!非常感谢。
    • 欢迎接受答案..如果你喜欢:)
    • 当然,但是你回答得太快了……所以你需要再等 5 分钟 :)
    • 哦,对不起!不知道。还有一件事,你能检查一下 k 的其他值(不是 3)
    • 是的,它似乎适用于任何整数 k。