获取标量场的级别（轮廓）（Matplotlib / Python）

Question

我想知道如何获得等高线图每个级别的顶点。

例如，我有这样的情节：

我猜它有超过 10 个级别。

我正在尝试获取每个顶点。

到目前为止，我有这段代码：

def iso_contours(scalar_fields):
    for scalar_field in scalar_fields:
        cs = plt.contour(scalar_field)
        paths = cs.collections[0].get_paths()
        print len(path)
        for path in paths:
            # Get vertices

但是，我只有一条路。如何实现我想要的正确方法？

提前谢谢你。

Answer 1

使用path.vertices:

import matplotlib.pyplot as plt
import numpy as np

x, y = np.meshgrid(np.linspace(0, 5, 100), np.linspace(0, 5, 100))
f = x ** y
g = y ** x
cs = plt.contour(x, y, (f - g))
for collection in cs.collections:
    paths = collection.get_paths()
    for path in paths:
        print(path.vertices.shape)

plt.show()

产量

(7, 2)
(28, 2)
(51, 2)
(172, 2)
(154, 2)
(51, 2)
(28, 2)
(7, 2)