重新排列具有不同長度的 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])