python - 在Python中将元组元素切片并堆叠到矩阵中

Question

我有一个元组列表(特征值、特征向量)，我必须从中提取前 60 个特征向量并将它们水平堆叠到一个矩阵中。我正在使用这段代码:

matrix_60 = np.hstack((eig_pairs[:60][1]).reshape(samples,1))

代码无法正常工作(我知道元组没有属性reshape)，但我无法找出正确的语法。预期结果是获取前 60 个特征向量并将它们按列堆叠(因此矩阵将具有 N 行 x 60 列)。

Answer 1

您可以使用np.stack使用 axis=1 按列堆叠向量:

np.stack([vec for val, vec in eig_pairs[:60]], axis=1)

列表推导将从前 60 个元组中提取特征向量。 stack 创建一个新轴，以便在您指定的维度中串联。

表达式eig_pairs[:60]仍然是一个列表，因此eig_pairs[:60][1]只是该列表中的第二个元组，而不是第二个每个元组中的元素。这就是为什么您需要理解来提取向量。