主页 / 问题 / 1132693
Accepted
Manifesto Mr
Asked:2020-05-28 16:37:38 +0800 CST2020-05-28 16:37:38 +0800 CST

形状不匹配错误

  • 772

我有一个数组

[[0 0 0 ... 0 0 0] 
[0 0 0 ... 0 0 0] 
[0 0 0 ... 0 0 0] 
... 
[0 0 0 ... 0 0 0]
[0 0 0 ... 0 0 0] 
[0 0 0 ... 0 0 0]]

零之间还有更多的数字。当我将此数组输入我的代码时，我遇到了一个错误：形状不匹配：对象无法广播到单个形状。这是我正在使用的代码：

z = b_line #b_line представляет из себя массив указанный выше
x = np.outer(np.linspace(0, len(z), len(z[0])), np.ones(len(z[0])))
y = x.copy().T # transpose
z = np.array(z)

fig = plt.figure()
ax = plt.axes(projection='3d')

ax.plot_surface(x, y, z,cmap='viridis', edgecolor='none')
ax.set_title('Surface plot')
plt.show()
python

2 个回答

  1. 为什么不提供可重现的代码片段？我们只能猜测您在那里写了什么或忘记写什么。

    由于您对我们隐藏了代码，因此强烈怀疑您忘记插入

    from mpl_toolkits.mplot3d import Axes3D

    这就是你的代码对我来说的样子：

    import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
b_line=[[0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0]]
z = b_line #b_line представляет из себя массив указанный выше
x = np.outer(np.linspace(0, len(z), len(z[0])), np.ones(len(z[0])))
y = x.copy().T # transpose
z = np.array(z)
fig = plt.figure()
ax = plt.axes(projection='3d')
ax.plot_surface(x, y, z,edgecolor='none',cmap='viridis')
ax.set_title('Surface plot')
plt.show()

    结果如下：在此处输入图像描述

    如您所见，一切正常。

    • 0
  2. Best Answer

    经过长时间的讨论，我可以提供以下经典版本，适用于任意矩阵

    import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
b_line=[
[1, 0, 0, 0, 0, 0],
[0, 1, 0, 0, 0, 0],
[0, 0, 1, 0, 0, 0],
[0, 0, 0, 1, 0, 0],
[0, 0, 0, 0, 1, 0],
[0, 0, 0, 0, 0, 1],
[1, 0, 0, 0, 0, 0],
[0, 1, 0, 0, 0, 0],
[0, 0, 1, 0, 0, 0],
[0, 0, 0, 1, 0, 0]
]
z = b_line #b_line представляет из себя массив указанный выше
x=np.linspace(0, len(z), len(z[0]))
y=np.linspace(0, len(z[0]),len(z))
X,Y=np.meshgrid(x,y)
z = np.array(z)
fig = plt.figure()
ax = plt.axes(projection='3d')
ax.plot_surface(X, Y, z)
ax.set_title('Surface plot')
plt.show()

    结果：在此处输入图像描述

    • 0

相关问题