contourpy_2d_contour_lines_with_matplotlib_visualization.py

1import contourpy
2import matplotlib.pyplot as plt
3import numpy as np
4
5# Create data to contour
6x, y = np.meshgrid(np.arange(7), np.arange(6))
7z = np.sin(x*np.pi/6.0) + np.cos(y*np.pi/3.0)
8
9# Create contour generator
10cont_gen = contourpy.contour_generator(x, y, z)
11
12# Calculate contour lines at level 0.5
13lines = cont_gen.lines(0.5)
14
15# Plot results
16fig, ax = plt.subplots()
17for line in lines:
18    ax.plot(line[:, 0], line[:, 1], color="red", linewidth=2)
19
20ax.set_title("contourpy quickstart")
21plt.show()