import matplotlib.pyplot as plt
import numpy as np
import os

script_path = os.path.abspath(__file__)
script_dir = os.path.dirname(script_path)
os.chdir(script_dir)


cmap = plt.colormaps['viridis']


fig, ax = plt.subplots(figsize=(2, 6))  
norm = plt.Normalize(vmin=0, vmax=10000)
fig.colorbar(plt.cm.ScalarMappable(norm=norm, cmap=cmap), cax=ax, orientation='vertical')


output_path = 'colorbar_purple_yellow.png'
plt.savefig(output_path, bbox_inches='tight', dpi=300)
plt.close()

output_path