用其他 numpy 數組填充 numpy 數組

Question

我有以下 numpy arrays：

    whole = np.array(
        [1, 0, 3, 0, 6]
    )

    sparse = np.array(
        [9, 8]
    )

現在我想用稀疏數組中的項目按時間順序替換整個數組中的每個零。 在示例中，我想要的數組如下所示：

    merged = np.array(
        [1, 9, 3, 8, 6]
    )

我可以自己編寫一個小算法來解決這個問題，但如果有人知道解決這個問題的高效方法，我將非常感謝你的幫助！

Answer 1

您是否假設 sparse 的長度與整個零的長度相同？

如果是這樣，您可以這樣做：

import numpy as np
from copy import copy

whole = np.array([1, 0, 3, 0, 6])
sparse = np.array([9, 8])

merge = copy(whole)
merge[whole == 0] = sparse

如果長度不匹配，則必須使用 len(...) 和切片將其限制為正確的長度。