Гладкий контурный участок полностью заполнен - PullRequest
Новая
       34

Гладкий контурный участок полностью заполнен

0 голосов
/ 04 октября 2019

У меня есть данные со значениями (X, Y, Z). Я попытался сделать график плотности со значениями Z для интенсивности. Однако график, который я получаю, не является гладким и имеет многогранник, то есть не полностью заполненный.

Ниже приведен код с Данные

, но я хочу получить плавный иполностью заполненный участок 


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

Это то, чего я хочу добиться с помощью contourplotf:

enter image description here

1 Ответ

1 голос
/ 04 октября 2019

plt.contourf() здесь не главная проблема, она просто работает с имеющимися данными. Проблема заключается в линейной интерполяции в scipy.interpolate.griddata().

. Я рекомендую не использовать griddata, а вместо этого использовать один из следующих методов:

Все эти методы будут заполненысетки. Если вы нанесете результат с помощью plt.imshow(), вы получите тип графика, который вы показываете в своем вопросе - это не plt.contourf() график.

Вот демонстрационный блокнот , показывающий всеиз этих подходов (включая griddata).

...