如何同步两个不同长度的数组？

Question

让我们考虑两个包含索引的数组：

x = [0,1,2,3,4,5...]
y = [0,3,6,9,12,...]

这些数组的长度可能略有不同，大约最多 3 个索引。 在此示例中，我们假设 len(x) = len(y) - 1 我想返回同步的 x，它将被那个 1 条目扩展，以便这些数组仍然相互对应（x[n]=y[n]+3）。 我想出了使用 np.searchsorted 方法的想法，但是它不起作用：

def synchronize_array(self, arr: np.ndarray) -> np.ndarray:
    sync_idx = np.searchsorted(arr, BASE_ARR)
    sync_idx[sync_idx >= len(arr)] = len(arr) - 1
    return arr[sync_idx]

Sync_idx 在这种情况下是 [0, n-1, n-1, n-1, ...] 有什么方法可以同步这些数组吗？

Answer 1

不清楚你所说的同步是什么意思，但你可以迭代两个数组，用itertools.zip_longest填充最短的数组

from itertools import zip_longest

x = [0, 1, 2, 3, 4, 5]
y = [0, 3, 6, 9, 12]

xy = zip_longest(x, y, fillvalue=0)
print(list(xy))

哪个产生

[(0, 0), (1, 3), (2, 6), (3, 9), (4, 12), (5, 0)]

干杯！