Bachelor-Thesis_Furkan_Karakocaoglu/testumgebung/CrowdModelling/Assets/Depictions_Years/Scripts/Python/OpenHeatmap.py

import json
import matplotlib.pyplot as plt
import matplotlib.cm as cm
from matplotlib.colors import LinearSegmentedColormap 
from mpl_toolkits.mplot3d import Axes3D

LOADFILE = "Assets/Resources/Json/output.json"

with open(LOADFILE, 'r') as filehandle:
    data = json.load(filehandle)

x = data['x']
y = data['y']
z = data['z']
dz = data['dz']
color_values = data['color_values']
color_intens = data['color_intens']

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

img = ax.bar3d(x, y, z, 1, 1, dz, color=color_values, shade=False)

cmap = LinearSegmentedColormap.from_list(name='day1', colors=[(0.40,0.76,0.65), (0.11,0.62,0.47)])
# cmap = LinearSegmentedColormap.from_list(name='day2', colors=['grey', (0.99,0.55,0.38), (0.85,0.37,0.01)])
# cmap = LinearSegmentedColormap.from_list(name='day3', colors=['grey', (0.55,0.63,0.80), (0.46,0.44,0.70)])

color_map = cm.ScalarMappable(cmap=cmap)
color_map.set_array(color_intens)
fig.colorbar(color_map)

plt.show()