matplotlib 使用plot_Surface在3D图形中显示背景图像

nhn9ugyo  于 2023-03-13  发布在  其他
关注(0)|答案(1)|浏览(285)

我正在寻找一种方法来显示一个.png图像的3D图形的背景。我尝试了this Post这里,但即使我复制的确切代码:

from mpl_toolkits.mplot3d import Axes3D
from matplotlib import cm
from matplotlib.ticker import LinearLocator, FormatStrFormatter
import matplotlib.pyplot as plt
import numpy as np
from matplotlib._png import read_png
from matplotlib.cbook import get_sample_data

fig = plt.figure()
ax = fig.gca(projection='3d')
X = np.arange(-5, 5, .25)
Y = np.arange(-5, 5, .25)
X, Y = np.meshgrid(X, Y)
R = np.sqrt(X**2 + Y**2)
Z = np.sin(R)
surf = ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=cm.winter,
                       linewidth=0, antialiased=True)

ax.set_zlim(-2.01, 1.01)
ax.zaxis.set_major_locator(LinearLocator(10))
ax.zaxis.set_major_formatter(FormatStrFormatter('%.02f'))

fn = get_sample_data("./grace_hopper.png", asfileobj=False)
arr = read_png(fn)
# 10 is equal length of x and y axises of your surface
stepX, stepY = 10. / arr.shape[0], 10. / arr.shape[1]

X1 = np.arange(-5, 5, stepX)
Y1 = np.arange(-5, 5, stepY)
X1, Y1 = np.meshgrid(X1, Y1)
Z = 
# stride args allows to determine image quality 
# stride = 1 work slow
ax.plot_surface(X1, Y1, 2.0, rstride=1, cstride=1, facecolors=arr)

plt.show()

我总是得到这个错误:

Traceback (most recent call last):
  File "c:\Users\XXX\ZeichnenFabrik\readTextFile.py", line 161, in <module>
    main()
  File "c:\Users\XXX\ZeichnenFabrik\readTextFile.py", line 157, in main
    plotGraph(nodedict,slines)
  File "c:\Users\XXX\ZeichnenFabrik\readTextFile.py", line 145, in plotGraph
    ax.plot_surface(X1, Y1, 0)
  File "C:\Program Files\Python36\lib\site-packages\mpl_toolkits\mplot3d\axes3d.py", line 1609, in plot_surface
    if Z.ndim != 2:
AttributeError: 'int' object has no attribute 'ndim'

我有办法解决吗?

ecfdbz9o

ecfdbz9o1#

看起来matplotlib中有一个变化--plot_surface的第三个参数必须是一个2D数组,所以用np.atleast_2d Package 常量:

ax.plot_surface(X1, Y1, np.atleast_2d(-2.0), rstride=10, cstride=10, facecolors=arr)

另请注意,arr.shape(600, 512, 3),而X1.shape(512, 600)。这会导致形状不匹配,从而产生IndexError。要避免此问题,请交换stepXstepY的定义:
x一个一个一个一个x一个一个二个x

相关问题