如何平滑matplotlib等高线图?

时间:2012-09-05 04:32:35

标签: python matplotlib

我有这种形状的numpy数组:(33,10)。当我绘制轮廓时,我会得到这样丑陋的图像: enter image description here

虽然contour()似乎没有关于平滑或某种插值特征的任何争论。

我不知何故预计提供等高线图的工具也应该提供平滑 在MPL中有直接的方法吗?

3 个答案:

答案 0 :(得分:55)

正如其他人已经指出的那样,您需要插入数据。

有许多不同的方法可以做到这一点,但对于初学者,请考虑scipy.ndimage.zoom

作为一个快速的例子:

import numpy as np
import scipy.ndimage
import matplotlib.pyplot as plt

data = np.loadtxt('data.txt')

# Resample your data grid by a factor of 3 using cubic spline interpolation.
data = scipy.ndimage.zoom(data, 3)

plt.contour(data)
plt.show()

enter image description here

答案 1 :(得分:8)

没有简单的方法来获得平滑的轮廓。另一种方法是尝试imshow。您可以查看here其他可能性。

import pylab as plt
import numpy as np

Z=np.loadtxt('data.txt')
plt.subplot(131)
plt.imshow(Z,interpolation='nearest')

plt.subplot(132)
plt.imshow(Z)

plt.subplot(133)
plt.imshow(Z,interpolation='gaussian')

plt.show()

enter image description here

答案 2 :(得分:3)

尝试使用gaussian_filter平滑数据集。有关详细信息,请参阅example