How can I plot a mathematical expression of two variables in python?

Question

I have a time- and depth-varying function, simplified to:

def f(z,t):
    return np.exp(-z)*np.sin(t-z)

z = np.linspace(0,3000,num=3001)
t = np.arange(0,40000,4000)

I want to plot the the result for all z at each time step in t, resulting in something like this:

enter image description here

but I don't know how. I'm sure it's very simple, but I'm not used to doing this in python.

unutbu · Accepted Answer · 2012-12-11 02:58:20Z

12

import matplotlib.pyplot as plt
import numpy as np

def f(z,t):
    return np.exp(-z)*np.sin(t-z)

z = np.linspace(0,5,3001)
t = np.arange(0,40000,4000)

for tval in t:
    plt.plot(z, f(z, tval))
plt.show()

enter image description here

edited Dec 11, 2012 at 2:58

answered Dec 11, 2012 at 2:51

unutbu

886k197 gold badges1.9k silver badges1.7k bronze badges

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

2 Comments

a different ben Over a year ago

ha, i feel like such a numpty. Thanks!

unutbu Over a year ago

well, now you know how to do it, and that's all that's important.

Collectives™ on Stack Overflow

How can I plot a mathematical expression of two variables in python?

1 Answer 1

2 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related