File: quiver_demo2.py

package info (click to toggle)
scitools 0.7-1
  • links: PTS, VCS
  • area: main
  • in suites: squeeze
  • size: 17,252 kB
  • ctags: 2,871
  • sloc: python: 28,744; sh: 112; makefile: 15
file content (40 lines) | stat: -rw-r--r-- 823 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
#!/usr/bin/env python

"""
Demonstration of the quiver command in combination with other plotting
commands.
"""

from scitools.std import *

setp(interactive=False)

xv, yv = ndgrid(linspace(-5,5,81), linspace(-5,5,81))
values = sin(sqrt(xv**2 + yv**2))

pcolor(xv, yv, values, shading='interp')

# create a coarser grid for the gradient field:
xv, yv = ndgrid(linspace(-5,5,21), linspace(-5,5,21), sparse=True)
values = sin(sqrt(xv**2 + yv**2))

# compute the gradient field:
uu, vv = gradient(values)

hold('on')
quiver(xv, yv, uu, vv, 'filled', 'k', axis=[-6,6,-6,6])
show()

#hardcopy('quiver2a.eps')
#hardcopy('quiver2a.png')


figure()
contour(xv, yv, values, 15, hold=True)
quiver(xv, yv, uu, vv, axis=[-6,6,-6,6])
show()

#hardcopy('quiver2b.eps')
#hardcopy('quiver2b.png')

raw_input("Press Return key to quit: ")