Can't Interpolate data in Pandas Pivot table
I'm trying to create a heatmap similar to:
But when I try this in Python code as below using Pandas interpolate on the pivot table I get errors:
import pandas as pd
import numpy as np
import matplotlib
import matplotlib.pyplot as plt
from matplotlib import rcParams
import scipy as sp
import scipy.ndimage
from scipy import interpolate
FILE_PATH = "." def load_nut_data(file_path=FILE_PATH):
csv_path = os.path.join(file_path, "some_file.csv")
return pd.read_csv(csv_path, usecols = [ 'GPSLon', 'GPSLat','Count'])
nut = load_nut_data() nut.head()
Z = nut.pivot_table(index='GPSLon', columns='GPSLat',
values='Count').T.values
X_unique = np.sort(nut.GPSLon.unique()) Y_unique =
np.sort(nut.GPSLat.unique()) X, Y = np.meshgrid(X_unique, Y_unique)
figure = plt.figure() ax = figure.add_subplot(111) contour =
ax.contourf(X, Y, Z, cmap=plt.cm.jet)
plt.show()
Any help or tips would be appreciated.
from Recent Questions - Stack Overflow https://ift.tt/3pVN0vg
https://ift.tt/3pLbKGA
Comments
Post a Comment