How to make a 3D data surface plot using matplotlib in python

Question

I am trying to make a 3d surface plot of experimental data using matplotlib. I would like to plot different Z values against the same X and Y axes. When I try the simple code below, I get the error

"plot_surface() missing 1 required positional argument: 'Z' ".

It seems that the Axes3D package only work if Z is given as a function of X and Y, rather than an actual data matrix. Does anybody know a way around this?

Please note that the Zmatrix that I need is actual data, but I just used a random matrix for illustration here.

import numpy as np
from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
fig = plt.figure()

X=[2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20]
Y= [0,2500,5000,7500,10000,15000,20000,25000,30000,35000,40000,45000,50000,55000,60000,65000,70000]

Zmatrix=np.random.rand(len(X),len(Y))


Axes3D.plot_surface(X,Y,Zmatrix)

lebelinoz · Accepted Answer · 2021-09-29 22:10:56Z

2

There were sone issues with your code: First you have to get a meshgrid of X and Y (all combinations as matrices). Next swap len(X) and len(Y) inside the Zmatrix. And first define ax = Axes3D(plt.gcf()) and plot_surface afterwards on ax.

import numpy as np
from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
fig = plt.figure()

X=[2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20]
Y= [0,2500,5000,7500,10000,15000,20000,25000,30000,35000,40000,45000,50000,55000,60000,65000,70000]
Xm, Ym = np.meshgrid(X, Y)
Zmatrix=np.random.rand(len(Y),len(X))

ax = Axes3D(plt.gcf())
ax.plot_surface(Xm, Ym, Zmatrix)

edited Sep 29, 2021 at 22:10

lebelinoz

5,09611 gold badges39 silver badges61 bronze badges

answered Mar 29, 2018 at 17:26

ilja

2,7022 gold badges18 silver badges23 bronze badges

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

Comments

Ranjeet · Accepted Answer · 2018-03-29 17:26:10Z

0

Here is an example of surface plot.

  import numpy as np
  from mpl_toolkits.mplot3d import Axes3D
  import matplotlib.pyplot as plt
  import random

  def fun(x, y):
      return x**2 + y

  fig = plt.figure()
  ax = fig.add_subplot(111, projection='3d')
  x = y = np.arange(-3.0, 3.0, 0.05)
  X, Y = np.meshgrid(x, y)
  zs = np.array([fun(x,y) for x,y in zip(np.ravel(X), np.ravel(Y))])
  Z = zs.reshape(X.shape)

  ax.plot_surface(X, Y, Z)

  ax.set_xlabel('X Label')
  ax.set_ylabel('Y Label')
  ax.set_zlabel('Z Label')

  plt.show()

answered Mar 29, 2018 at 17:26

Ranjeet

3922 gold badges4 silver badges13 bronze badges

Collectives™ on Stack Overflow

How to make a 3D data surface plot using matplotlib in python

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related