从matplotlib中的轮廓获取坐标?

时间:2013-10-17 05:28:32

标签: python matplotlib

背景

the documentation example here,可以使用代码段轻松生成以下等高线图。

import matplotlib
import numpy as np
import matplotlib.cm as cm
import matplotlib.mlab as mlab
import matplotlib.pyplot as plt

matplotlib.rcParams['xtick.direction'] = 'out'
matplotlib.rcParams['ytick.direction'] = 'out'

delta = 0.025
x = np.arange(-3.0, 3.0, delta)
y = np.arange(-2.0, 2.0, delta)
X, Y = np.meshgrid(x, y)
Z1 = mlab.bivariate_normal(X, Y, 1.0, 1.0, 0.0, 0.0)
Z2 = mlab.bivariate_normal(X, Y, 1.5, 0.5, 1, 1)
# difference of Gaussians
Z = 10.0 * (Z2 - Z1)

# Create a simple contour plot with labels using default colors.  The
# inline argument to clabel will control whether the labels are draw
# over the line segments of the contour, removing the lines beneath
# the label
plt.figure()
CS = plt.contour(X, Y, Z)
plt.clabel(CS, inline=1, fontsize=10)
plt.title('Simplest default with labels')

enter image description here

我的目标

我已经获得了轮廓图,同时获得了matplotlib.contour.QuadContourSet实例CS。在示例代码段中,CS仅用于clabel()。但是对于我的情况,我需要获得轮廓线的方程或坐标集以进行进一步计算。

如何从实例CS中提取轮廓线的坐标?要么 我怎样才能以其他方式实现它?

我打赌必须有这样做的方法。否则,轮廓事物只是一个“可视化的花瓶”。

1 个答案:

答案 0 :(得分:11)

您可以从CS.allsegs列表中获取轮廓的坐标。

尝试:

dat0= CS.allsegs[0][0]
plt.plot(dat0[:,0],dat0[:,1])

绘制第一个(-1)轮廓水平。