Issue in plotting 3d surface in python

Question

I have the following data, trying to plot a 3d surface.

data = [[10, 10, 0.84496124031007758],
    [10, 20, 0.87209302325581395],
    [10, 30, 0.88139534883720927],
    [20, 10, 0.86201550387596892],
    [20, 20, 0.87441860465116272],
    [20, 30, 0.88992248062015500],
    [30, 10, 0.87984496124031009],
    [30, 20, 0.89922480620155043],
    [30, 30, 0.92015503875968996]]

x, y, z = zip(*data)
fig = plt.figure()
ax = plt.axes(projection='3d')
ax.plot_surface(x, y, z,cmap='viridis', edgecolor='none')
plt.show()

I am retrieving below error,

File "C:/Users/40227422/PycharmProjects/VideoDetection.py", line 29, in <module>
   ax.plot_surface(x, y, z,cmap='viridis', edgecolor='none')
File "C:\Users\40227422\AppData\Local\Continuum\miniconda3\envs\tensorflow\lib\site- 
packages\mpl_toolkits\mplot3d\axes3d.py", line 1496, in plot_surface
   if Z.ndim != 2:
AttributeError: 'tuple' object has no attribute 'ndim'

Thanks in advance

Ivan · Accepted Answer · 2020-12-09 17:16:44Z

1

The first three arguments of Axes3D.plot_surface all need to be 2D arrays.

Replacing your call with the following will work:

ax.plot_surface(np.array(x).reshape(3,3), np.array(y).reshape(3,3), np.array(z).reshape(3,3), cmap='viridis', edgecolor='none')

Or with a list comprehension + destructuring:

ax.plot_surface(*[np.array(d).reshape(3,3) for d in zip(*data)], cmap='viridis', edgecolor='none')

answered Dec 9, 2020 at 17:16

Ivan

41.3k9 gold badges78 silver badges120 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Issue in plotting 3d surface in python

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related