Numpy 连接，使用 * 或类似的

Question

我有一个 numpy 数组列表。像这样的东西（这不是同一个例子，而是相似的）lst = [np.array([ 1,2,3,4,5,6 ]).reshape(-1, 1), np.array([ 1,2,3,4,5,6 ]).reshape(-1, 1), np.array([ 1,2,3,4,5,6 ]).reshape(-1, 1)]在这种情况下，我lst有 3 个 numpy 数组，它们的形状是 (6,1)，现在我想将它连接起来，如下所示：# array([[1, 1, 1],#        [2, 2, 2],#        [3, 3, 3],#        [4, 4, 4],#        [5, 5, 5],#        [6, 6, 6]])这完美地做到了这一点......example = np.c_[lst[0], lst[1], lst[2]]但我的 lst 并不总是相同的大小，所以我尝试了这个。example = np.c_[*lst]但它不起作用。有没有办法以这种方式连接整个列表？

Answer 1

您可以使用column_stack功能：

import numpy as np

lst = [np.array([1, 2, 3, 4, 5, 6]).reshape(-1, 1), np.array([1, 2, 3, 4, 5, 6]).reshape(-1, 1), np.array([1, 2, 3, 4, 5, 6]).reshape(-1, 1)]

example = np.column_stack(lst)

print(example)

[[1 1 1]

 [2 2 2]

 [3 3 3]

 [4 4 4]

 [5 5 5]

 [6 6 6]]