How can I get the (x,y) values of a line that is plotted by a contour plot (Matplotlib)?

MatplotlibPythonData Visualization

To get the (x, y) values of a line that is plotted by a contour plot, we can take the following steps −

  • Set the figure size and adjust the padding between and around the subplots.
  • Create a 3D contour plot using contour() method.
  • Get the contour plot collections and get the paths.
  • To display the figure, use show() method.

Example

import matplotlib.pyplot as plt

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

m = [[3, 2, 1, 0], [2, 4, 1, 0], [2, 4, 1, 3], [4, 3, 1, 3]]
cs = plt.contour([3, 4, 2, 1], [5, 1, 2, 3], m)
p1 = cs.collections[0].get_paths()

for item in p1:
   print(item.vertices)

plt.show()

Output

raja
Published on 05-Jun-2021 08:09:20
Advertisements