平滑的Contourf图已完全填充

时间:2019-10-03 23:18:29

标签: python matplotlib contourf

我有(X,Y,Z)个值的数据。我试图用Z值绘制强度密度图。但是,我得到的图并不平滑,并且具有多面体,即未完全填充。

以下是带有Data

的代码

但是我想获得平滑且完全填充的图


import numpy as np
from scipy.interpolate import griddata
import matplotlib.pyplot as plt
import xlrd
location = "~/Desktop/Data.xlsx"
data = xlrd.open_workbook(location)
sheet = data.sheet_by_index(0)
sample=2000

x=np.array(sheet.col_values(0))[0:sample]
y=np.array(sheet.col_values(1))[0:sample]
z=np.hamming(9000)[0:sample]
print z
def plot_contour(x,y,z,resolution = 500,contour_method='cubic'):
    resolution = str(resolution)+'j'
    X,Y = np.mgrid[min(x):max(x):complex(resolution),   min(y):max(y):complex(resolution)]
    points = [[a,b] for a,b in zip(x,y)]
    Z = griddata(points, z, (X, Y), method=contour_method)
    return X,Y,Z

X,Y,Z = plot_contour(x,y,z,resolution = 500,contour_method='linear')

plt.style.context("seaborn-deep")

plt.contourf(X,Y,Z)
plt.colorbar()
plt.show()

这是输出:

enter image description here

这是我要使用轮廓线图实现的目标:

enter image description here

1 个答案:

答案 0 :(得分:1)

plt.contourf()并不是这里的主要问题,它只是处理它拥有的数据。问题是scipy.interpolate.griddata()中的线性插值。

我建议不要使用griddata,而应使用以下方法之一:

所有这些方法将填入网格。如果您用plt.imshow()绘制结果,您将得到问题中显示的图表类型-不是plt.contourf()图表。

Here's a demo notebook显示了所有这些方法(包括griddata)。