我有(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()
这是输出:
这是我要使用轮廓线图实现的目标:
答案 0 :(得分:1)
plt.contourf()
并不是这里的主要问题,它只是处理它拥有的数据。问题是scipy.interpolate.griddata()
中的线性插值。
我建议不要使用griddata
,而应使用以下方法之一:
scipy.interpolate.Rbf()
-这是您以前使用的(see my previous answer)。verde
—一个很棒的网格软件包。sklearn.gaussian_process
-或其他一些预测模型。所有这些方法将填入网格。如果您用plt.imshow()
绘制结果,您将得到问题中显示的图表类型-不是plt.contourf()
图表。
Here's a demo notebook显示了所有这些方法(包括griddata
)。