matplotlib — извлечение значений из контурных линий

Эта пара вопрос/ответ показывает, как извлекать вершины из контурного графика:

p = cs.collections[0].get_paths()[0]
v = p.vertices
x = v[:,0]
y = v[:,1]

Но как получить значение (например, z для модели высот) для каждого пути?


python matplotlib contour
person WKT    schedule 11.06.2013    source источник

Ответы (1)


arrow_upward
3
arrow_downward

Прямого пути нет, но cs.collections находится в том же порядке, что и cs.levels (это значения "z", которые вам нужны).

Поэтому проще всего сделать что-то вроде:

lookup = dict(zip(cs.collections, cs.levels))
z = lookup[line_collection_artist]

В качестве быстрого интерактивного примера:

import numpy as np
import matplotlib.pyplot as plt

def main():
    fig, ax = plt.subplots()
    cs = ax.contour(np.random.random((10,10)))

    callback = ContourCallback(cs)
    plt.setp(cs.collections, picker=5)
    fig.canvas.mpl_connect('pick_event', callback)

    plt.show()

class ContourCallback(object):
    def __init__(self, cs):
        self.lookup = dict(zip(cs.collections, cs.levels))

    def __call__(self, event):
        print self.lookup[event.artist]

main()
person Joe Kington    schedule 12.06.2013