交织两个numpy 1D数组的所有方法,保持每个数组的元素顺序

时间:2019-09-19 23:07:01

标签: python-3.x numpy-ndarray

考虑两个numpy数组 a b a 是整数1到5的一些排列,而 b 是整数-5到-1的一些排列。

a = [ 1  3  4  2  5]
b = [-4 -3 -5 -1 -2]

我想找到所有包含 a b 的10个值的数组 c ,但是 a b 元素必须保留。

有效c的示例:

c = [1 3 -4 -3 -5 4 2 -1 5 -2]

无效c的示例:(-5和-3违反了模式)

c = [1 3 -4 -5 -3 4 2 -1 5 -2]

我在stackoverflow中发现了很多情况,其中数组元素以交替的方式交错,但是在这里并不适用。一种可能的方法可能是生成所有5个0和5个1的一维数组,如下所示:

[0 0 1 1 1 0 0 1 0 1]

每个0表示'next' a 值,每个1表示'next' b 值。

我希望会有一种有效的方法来解决这个问题。

0 个答案:

没有答案