python - 如何使用numpy的hstack？

Question

我有一个大型 numpy.ndarray 数组，我想从中提取第 4 列和第 5 列并将这些列放入二维数组中。 [i,0] 元素应该是第 4 列的值，[i,1] 应该是第 5 列的元素。

我尝试使用 numpy.hstack 函数来执行此操作。

a = numpy.asarray([1, 2, 3, 4, 5])
for i in range(5):
    a = numpy.vstack([a, numpy.asarray([1, 2, 3, 4, 5])])

combined = np.hstack([a[:,3], a[:,4]])

但是，这只是给了我一个 nx1 数组。我已经尝试了多种使用连接的方法，看起来像这些例子:

combined = np.concatenate([a[:,3], a[:,4]])

combined = np.concatenate([a[:,3], a[:,4]], axis=1)

combined = np.concatenate([a[:,3].T, a[:,4].T])

我觉得 hstack 是我想要的功能，但我似乎无法弄清楚如何让它给我一个 nx2 数组。谁能指出我正确的方向？感谢您的帮助。

Answer 1

只需按如下方式切分您的数据:

X = [[0 1 2 3 4]
     [0 1 2 3 4]
     [0 1 2 3 4]
     [0 1 2 3 4]]

slicedX = X[:,3:5]

结果:

[[3 4]
 [3 4]
 [3 4]
 [3 4]]