重新排列具有不同长度的 2 个数组以创建第三个数组 Numpy Python

Question

我在a和b下方有 2 个数组组合来生成result 。 a乘以a_multiplier ， b乘以b_multiplier 。 a和b有不同长度的数组，我想将它们组合和相乘，然后按a*a_multiplier, b*b_multiplier,a*a_multiplier.....顺序重新排列它们。 我怎样才能修改result函数以便预期输出起作用？

import numpy as np 

a_multiplier = 3
b_multiplier = 5

a = np.array([5,32,1,4,3])
b = np.array([1,5,11,3])

result = np.vstack([a * a_multiplier, b * b_multiplier]).flatten("F")

预期输出：

[15  5 96 25  3 55 12 15 9]

值错误：

ValueError: all the input array dimensions for the concatenation axis must match exactly, but along dimension 1, the array at index 0 has size 5 and the array at index 1 has size 4

Answer 1

干得好：

import numpy as np

a_multiplier = 3
b_multiplier = 5

a = np.array([5,32,1,4,3])
b = np.array([1,5,11,3])

result = np.empty((a.size + b.size,), dtype=a.dtype)
result[0::2] = a * a_multiplier
result[1::2] = b * b_multiplier

输出：

array([15,  5, 96, 25,  3, 55, 12, 15,  9])