将列表转换为 numpy 矩阵

Question

我有一个矩阵X和一个列表centres。我想创建一个新的 numpy matrix td_X。

我想td_X用由X索引等于中心的行组成的填充。在伪代码中：

for each in centers:
  td_X.append(X[eacj])

我如何使用 Python 做到这一点？

Answer 1

由于该中间尺寸是尺寸 1，只需重塑或挤压它。

x = np.array(alist)
x = np.squeeze(x)  # or
x = x.reshape(45, 5785)  # or
x = x[:,0,:]

您可以删除列表中的额外嵌套层，但是使用数组进行这种重塑要容易得多。