无法从numpy数组中检索值

Question

我有一个numpy数组，如下所示，还有一个重塑：

X = M.dot(X1) + B #prints [[value1 value2 ]]
X.reshape(2,1)

然后，我试图将这一点与其他观点结合起来，如下所示：

plt.plot([X[0],...],[X[1],...],'-og','LineWidth',2)

但是，我无法检索任何值，索引1超出了范围。如何解决此问题并访问这两个值

Answer 1

注意：.reshape不在适当的位置工作，而是返回一些内容。您的原始X具有(1 ,2)形状。在plot语句中X[1]确实访问了索引为1的行（即“第二”行）——该行不存在

无论如何，试试看

X = X.squeeze()  # X.shape now is (2,)

这将X转换为1d数组。然后X[1]访问第二个元素

或者，您可以将plot语句更改为使用X[0, 1]而不是使用X[1]作为具有1行和2列的原始X