[Matplotlib-users] Multiplot with one colorbar

Discussion:

Dyah rahayu martiningrum

2015-03-06 09:36:44 UTC

Hello all,

I make multi plot with colorbars. I need help, how do make only one
colorbar for six panels? I also want to show only lowest x-axis. I copy my
recent code and figure here.

# Open file
fd = nc.Dataset(fname, 'r')

# Read variables
beam = fd.variables['beam'][:]
rng = fd.variables['range'][:]
tim = fd.variables['time'][:]
pwr = fd.variables['pwr'][:]
nfft = fd.variables['nfft'][0]
pn = fd.variables['pnoise'][:]
width=fd.variables['width'][:]
# Close netCDF file
fd.close()

# Specify beam
ibeam = ibeams[i]

# Time convertion
tim = tim/3600.0

#Plot
p_plot = pwr[ibeam]
for it in range(len(tim)):
p_plot[it] = p_plot[it] - pn[ibeam][it] - 10.*np.log10(nfft)
p_plot = p_plot.transpose()

#Specify subplot
pl.subplot(611 + i)
#Contour plot
v= np.linspace(5., 50., 5., endpoint=True)
x= pl.contourf(tim, rng, p_plot, v, cmap=pl.cm.jet)
x = pl.colorbar(ticks=v)
x.ax.set_title('dB')
print x

# Set X and Y axis lower/uppser limit
set_xy = range(4)
set_xy[0] = 18.0 # x min
set_xy[1] = 30.0 # x max
set_xy[2] = 200.0 # y min
set_xy[3] = 550.0 # y max
pl.axis(set_xy)

# Set labels
pl.xlabel('time (hours)')
pl.ylabel('range (km)')
pl.show()

Thank you....

Arnaldo Russo

2015-03-06 13:39:58 UTC

Permalink

Hi,
.
Your example is not much clear. It's better post some short example and a
running piece of code.
btw, maybe your problem should be solved with this post:
http://stackoverflow.com/questions/13784201/matplotlib-2-subplots-1-colorbar

Cheers,
Arnaldo

---
*Arnaldo D'Amaral Pereira Granja Russo*
Lab. de Estudos dos Oceanos e Clima
Instituto de Oceanografia - FURG

Post by Dyah rahayu martiningrum
Hello all,
I make multi plot with colorbars. I need help, how do make only one
colorbar for six panels? I also want to show only lowest x-axis. I copy my
recent code and figure here.
# Open file
fd = nc.Dataset(fname, 'r')
# Read variables
beam = fd.variables['beam'][:]
rng = fd.variables['range'][:]
tim = fd.variables['time'][:]
pwr = fd.variables['pwr'][:]
nfft = fd.variables['nfft'][0]
pn = fd.variables['pnoise'][:]
width=fd.variables['width'][:]
# Close netCDF file
fd.close()
# Specify beam
ibeam = ibeams[i]
# Time convertion
tim = tim/3600.0
#Plot
p_plot = pwr[ibeam]
p_plot[it] = p_plot[it] - pn[ibeam][it] - 10.*np.log10(nfft)
p_plot = p_plot.transpose()
#Specify subplot
pl.subplot(611 + i)
#Contour plot
v= np.linspace(5., 50., 5., endpoint=True)
x= pl.contourf(tim, rng, p_plot, v, cmap=pl.cm.jet)
x = pl.colorbar(ticks=v)
x.ax.set_title('dB')
print x
# Set X and Y axis lower/uppser limit
set_xy = range(4)
set_xy[0] = 18.0 # x min
set_xy[1] = 30.0 # x max
set_xy[2] = 200.0 # y min
set_xy[3] = 550.0 # y max
pl.axis(set_xy)
# Set labels
pl.xlabel('time (hours)')
pl.ylabel('range (km)')
pl.show()
Thank you....
------------------------------------------------------------------------------
Dive into the World of Parallel Programming The Go Parallel Website,
sponsored
by Intel and developed in partnership with Slashdot Media, is your hub for
all
things parallel software development, from weekly thought leadership blogs
to
news, videos, case studies, tutorials and more. Take a look and join the
conversation now. http://goparallel.sourceforge.net/
_______________________________________________
Matplotlib-users mailing list
https://lists.sourceforge.net/lists/listinfo/matplotlib-users

Eric Firing

2015-03-06 17:39:30 UTC

Permalink

An old example of something like this is here:

http://matplotlib.org/examples/pylab_examples/multi_image.html

Eric

Jody Klymak

2015-03-06 20:21:50 UTC

Permalink

Hi,

I guess I don't understand the "[axx for axx in ax.flat]" command, but this steals from all the axes.

Cheers, Jody

fig,ax = plt.subplots(2,2)
for i in range(2):
for j in range(2):
im=ax[i,j].imshow(np.ones((20,20)))
im.set_clim([-1.,2.])
cax,kw = mpl.colorbar.make_axes([axx for axx in ax.flat],shrink=0.5)
plt.colorbar(im, cax=cax, **kw)

Post by Eric Firing

http://matplotlib.org/examples/pylab_examples/multi_image.html
Eric
------------------------------------------------------------------------------
Dive into the World of Parallel Programming The Go Parallel Website, sponsored
by Intel and developed in partnership with Slashdot Media, is your hub for all
things parallel software development, from weekly thought leadership blogs to
news, videos, case studies, tutorials and more. Take a look and join the
conversation now. http://goparallel.sourceforge.net/
_______________________________________________
Matplotlib-users mailing list
https://lists.sourceforge.net/lists/listinfo/matplotlib-users

Dyah rahayu martiningrum

2015-03-12 02:43:05 UTC

Permalink

Thank you so much Jody, Eric, Arnaldo, and Joy.
I will try your suggestion.

Dyah

Post by Jody Klymak
Hi,
I guess I don't understand the "[axx for axx in ax.flat]" command, but
this steals from all the axes.
Cheers, Jody
fig,ax = plt.subplots(2,2)
im=ax[i,j].imshow(np.ones((20,20)))
im.set_clim([-1.,2.])
cax,kw = mpl.colorbar.make_axes([axx for axx in ax.flat],shrink=0.5)
plt.colorbar(im, cax=cax, **kw)

Post by Eric Firing

http://matplotlib.org/examples/pylab_examples/multi_image.html
Eric

------------------------------------------------------------------------------

Post by Eric Firing
Dive into the World of Parallel Programming The Go Parallel Website,

sponsored

Post by Eric Firing
by Intel and developed in partnership with Slashdot Media, is your hub

for all

Post by Eric Firing
things parallel software development, from weekly thought leadership

blogs to

Post by Eric Firing
news, videos, case studies, tutorials and more. Take a look and join the
conversation now. http://goparallel.sourceforge.net/
_______________________________________________
Matplotlib-users mailing list
https://lists.sourceforge.net/lists/listinfo/matplotlib-users

--
===============================================
Dyah R Martiningrum
Student at Department of Communications and Computer Engineering
Graduate School of Informatics, Kyoto University
RISH, Gokasho, Uji shi, Kyoto, 611-0011, Japan.
Office : +81 0774 38 3856
Mobile : +81 090 6676 1981
===============================================

Jody Klymak

2015-04-01 17:24:45 UTC

Permalink

Hi all,

Just to follow up, I wrote the following helper function to put a single colorbar to the right of a) an axes, b) a list of axes, c) to the right of a gridspec. I was getting a little peeved with colorbar not working as I liked with subplots or gridspec, and this kind of does the trick for me for most of my use cases.

Im probably not the worlds best python programmer so feel free to make clearer.

Cheers, Jody

def colorbarRight(pcm,ax,fig,shrink=0.7,width=0.025,gap=0.03,**kwargs):
'''
def colorbarRight(pcm,ax,fig,shrink=0.7,width=0.05,gap=0.02)

Position colorbar to the right of axis 'ax' with colors from artist pcm.
ax can be an array of axes such as that returned by "subplots".

ax can also be a GridSpec, in which case the colorbar is centered to the
right of the grid.

Defaults might no leave enough room for the colorbar on the right side, so
you should probably use subplots_adjust() or gridspec_update() to make more
space to the right:

# with subplots:
import matplotlib.pyplot as plt
fig,ax=plt.subplots(2,2)
fig.subplots_adjust(right=0.87)
for axx in ax.flatten():
pcm=axx.pcolormesh(rand(10,10))
colorbarRight(pcm,ax,fig,extend='max')

# with gridspec:
import matplotlib.gridspec
import matplotlib.pyplot as plt
fig=plt.figure()

gs = gridspec.GridSpec(2,2)
gs.update(right=0.87)
for ii in range(2):
for jj in range(2):
ax=plt.subplot(gs[ii,jj])
pcm=ax.pcolormesh(rand(10,10))
colorbarRight(pcm,gs,fig,extend='max')
'''
import numpy as np

if type(ax) is matplotlib.gridspec.GridSpec:
# gridspecs are different than axes:
pos = ax.get_grid_positions(fig)
y0 = pos[0][-1]
y1 = pos[1][0]
x1 = pos[3][-1]
else:
if ~(type(ax) is np.ndarray):
# these are supposedly axes:
ax=np.array(ax)
# get max x1, min y0 and max y1
y1 = 0.
y0 = 1.
x1=0.
for axx in ax.flatten():
pos=axx.get_position()
x1=np.max([pos.x1,x1])
y1=np.max([pos.y1,y1])
y0=np.min([pos.y0,y0])
height = y1-y0
pos2 = [x1 + gap, y0 + (1.-shrink)*height/2., width, height*shrink]
cax=axes(position=pos2)
fig.colorbar(pcm,cax=cax,**kwargs)

Post by Dyah rahayu martiningrum
Thank you so much Jody, Eric, Arnaldo, and Joy.
I will try your suggestion.
Dyah
Hi,
I guess I don't understand the "[axx for axx in ax.flat]" command, but this steals from all the axes.
Cheers, Jody
fig,ax = plt.subplots(2,2)
im=ax[i,j].imshow(np.ones((20,20)))
im.set_clim([-1.,2.])
cax,kw = mpl.colorbar.make_axes([axx for axx in ax.flat],shrink=0.5)
plt.colorbar(im, cax=cax, **kw)

http://matplotlib.org/examples/pylab_examples/multi_image.html <http://matplotlib.org/examples/pylab_examples/multi_image.html>
Eric
------------------------------------------------------------------------------
Dive into the World of Parallel Programming The Go Parallel Website, sponsored
by Intel and developed in partnership with Slashdot Media, is your hub for all
things parallel software development, from weekly thought leadership blogs to
news, videos, case studies, tutorials and more. Take a look and join the
conversation now. http://goparallel.sourceforge.net/ <http://goparallel.sourceforge.net/>
_______________________________________________
Matplotlib-users mailing list
https://lists.sourceforge.net/lists/listinfo/matplotlib-users <https://lists.sourceforge.net/lists/listinfo/matplotlib-users>

------------------------------------------------------------------------------
Dive into the World of Parallel Programming The Go Parallel Website, sponsored
by Intel and developed in partnership with Slashdot Media, is your hub for all
things parallel software development, from weekly thought leadership blogs to
news, videos, case studies, tutorials and more. Take a look and join the
conversation now. http://goparallel.sourceforge.net/ <http://goparallel.sourceforge.net/>
_______________________________________________
Matplotlib-users mailing list
https://lists.sourceforge.net/lists/listinfo/matplotlib-users <https://lists.sourceforge.net/lists/listinfo/matplotlib-users>
--
===============================================
Dyah R Martiningrum
Student at Department of Communications and Computer Engineering
Graduate School of Informatics, Kyoto University
RISH, Gokasho, Uji shi, Kyoto, 611-0011, Japan.
Office : +81 0774 38 3856
Mobile : +81 090 6676 1981
===============================================

--
Jody Klymak
http://web.uvic.ca/~jklymak/