使用matplotlib pcolor进行插值

时间:2012-08-22 21:51:06

标签: python numpy matplotlib scipy

我有两个numpy数组,第一个是(30,365)并且包含全年30个深度的值,第二个数组是(30,1)并且包含与之对应的实际深度(以米为单位)第一个数组中的深度。我想绘制第一个数组,以便根据第二个数组缩放深度,但我也希望对数据进行插值(前几个深度相对靠近在一起,而较低的深度相距较远,为pcolor图像提供了块状外观。)

这就是我正在做的事情:

import numpy as np
import matplotlib.pyplot as plt

x = np.arange(0, 365, 1)
X, Y = np.meshgrid(x, depth)    #depth is the (30, 1) array

plt.pcolor(X, -Y, data)         #data is the (30, 365) array

这导致了块状外观,关于如何获得更平滑的图形的任何想法?

2 个答案:

答案 0 :(得分:4)

您的深度是否在规则网格上(即恒定间距)?如果是,您可以使用imshow并使用extent关键字和aspect='auto'设置范围。否则,你有两个选择。

您可以改为使用pcolormesh并使用shading='gouraud'。这将有助于清晰的颜色量化,但不如插值。

第二种选择是将数据内插到新的常规深度网格,因此您可以使用imshow和不同的插值选项。例如,要仅沿深度方向进行插值,您可以使用scipy的interpolate.interp1d

from scipy.interpolate import interp1d

fint = interp1d(depth, data.T, kind='cubic')
newdata = fint(newdepth).T

添加了.T,因为插值必须在最后一个索引上,而深度是数据的第一个索引。如果您愿意,可以将kind替换为'linear'

答案 1 :(得分:3)

不,pcolor不进行插值。您可以尝试NonUniformImage甚至imshow。查看示例here