python - Numpy:如何将单个数组堆叠到更大数组的每一行中并将其转换为二维数组？

Question

我有一个名为 heartbeats 的 numpy 数组，有 100 行。每行有 5 个元素。

我还有一个名为 time_index 的数组，包含 5 个元素。我需要在每行心跳前添加时间索引。

heartbeats = np.array([
    [-0.58, -0.57, -0.55, -0.39, -0.40],
    [-0.31, -0.31, -0.32, -0.46, -0.46]
])
time_index = np.array([-2, -1, 0, 1, 2])

我需要什么:

array([-2, -0.58],
      [-1, -0.57],
      [0, -0.55],
      [1, -0.39],
      [2, -0.40],
      [-2, -0.31],
      [-1, -0.31],
      [0, -0.32],
      [1, -0.46],
      [2, -0.46])

我只写了两行心跳来说明。

Answer 1

假设您正在使用 numpy，您要查找的确切输出数组可以通过堆叠 time_index 的重复版本来制作。与 heartbeats 的困惑版本:

np.stack((np.tile(time_index, len(heartbeats)), heartbeats.ravel()), axis=-1)