在Python中重塑矩阵

Question

我目前正在尝试重塑大小为 10304 80 的图像矩阵。我试图将矩阵的前 5 列给我 10304 5 并将其重塑为 (112*92,5)。这是我的尝试，但我不断收到值错误无法将大小为 20608 的数组重塑为形状 (10304,5)。关于如何做到这一点有什么建议吗？

test_PC_first5 = test_PC[:, [1,5]]

test_PC_first5 = test_PC_first5.reshape(112*92,5)

plt.plot(test_PC_first5)

Answer 1

您的代码所做的只是获取第 1 列和第 5 列，这就是为什么您的形状为 20608，即 2*10304。你需要编辑你的第一行test_PC_first5 = test_PC[:, 0:5]为了保留前 5 行。请记住索引从 0 而不是 1 开始