-
Notifications
You must be signed in to change notification settings - Fork 0
/
colorbars_v0.py
49 lines (40 loc) · 1.78 KB
/
colorbars_v0.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
'''
This script contains some functions to add colorbars for various parameters
'''
import matplotlib.pyplot as plt
def addtempcolorbar(ax,fig,im,clevs):
'''
This function adds a new colorbar on its own axes for CAPE
Inputs: ax, fig are matplotlib axis/figure objects, im is the contourf object,
and clevs is the contour levels used in the contourf plot
Outputs: just call it and it'll put the colorbar in the right place
This code was adapted from Dr. Kim Wood's Community Tools repo
'''
axes_bbox = ax.get_position()
left = axes_bbox.x0
bottom = axes_bbox.y0 - 0.015
width = 0.38
height = 0.01
cax = fig.add_axes([left, bottom, width, height])
cbar = plt.colorbar(im, cax=cax, ticks=clevs, orientation='horizontal')
#cbar.ax.xaxis.set_ticks_position('top')
cbar.ax.tick_params(labelsize=8)
cbar.set_label('2m Temperature (F)', size=8) # MODIFY THIS for other fields!! MODIFY THIS for other fields!!
def addcloudcolorbar(ax,fig,im,clevs):
'''
This function adds a new colorbar on its own axes for reflectivity
Inputs: ax, fig are matplotlib axis/figure objects, im is the contourf object,
and clevs is the contour levels used in the contourf plot
Outputs: just call it and it'll put the colorbar in the right place
This code was adapted from Dr. Kim Wood's Community Tools repo
'''
axes_bbox = ax.get_position()
left = axes_bbox.x0 + 0.39
bottom = axes_bbox.y0 - 0.015
width = 0.38
height = 0.01
cax = fig.add_axes([left, bottom, width, height])
cbar = plt.colorbar(im, cax=cax, ticks=clevs, orientation='horizontal')
#cbar.ax.xaxis.set_ticks_position('top')
cbar.ax.tick_params(labelsize=8)
cbar.set_label('Total Cloud Cover (%)', size=8) # MODIFY THIS for other fields!!