Atur ukuran colorbar Matplotlib agar sesuai dengan grafik

Saya tidak bisa mendapatkan colorbar pada grafik imshow seperti ini sama tingginya dengan grafik, singkatnya menggunakan Photoshop. Bagaimana cara mendapatkan ketinggian yang cocok?

Anda dapat melakukan ini dengan mudah dengan AxisDivider matplotlib .

Contoh dari halaman tertaut juga berfungsi tanpa menggunakan subplot:

import matplotlib.pyplot as plt
from mpl_toolkits.axes_grid1 import make_axes_locatable
import numpy as np

plt.figure()
ax = plt.gca()
im = ax.imshow(np.arange(100).reshape((10,10)))

# create an axes on the right side of ax. The width of cax will be 5%
# of ax and the padding between cax and ax will be fixed at 0.05 inch.
divider = make_axes_locatable(ax)
cax = divider.append_axes("right", size="5%", pad=0.05)

plt.colorbar(im, cax=cax)

Kombinasi ini (dan nilai yang dekat dengan ini) tampaknya "ajaib" berfungsi bagi saya untuk menjaga colorbar diskalakan ke plot, tidak peduli berapa ukuran layar.

plt.colorbar(im,fraction=0.046, pad=0.04)

Ini juga tidak memerlukan berbagi sumbu yang dapat membuat plot keluar dari kotak.

@ bogatron sudah memberikan jawaban yang disarankan oleh docs matplotlib , yang menghasilkan ketinggian yang tepat, tetapi memperkenalkan masalah yang berbeda. Sekarang lebar colorbar (dan juga ruang antara colorbar dan plot) berubah dengan lebar plot. Dengan kata lain, rasio aspek dari colorbar tidak diperbaiki lagi.

Untuk mendapatkan ketinggian yang tepat dan rasio aspek tertentu, Anda harus menggali lebih dalam axes_grid1modul misterius .

import matplotlib.pyplot as plt
from mpl_toolkits.axes_grid1 import make_axes_locatable, axes_size
import numpy as np

aspect = 20
pad_fraction = 0.5

ax = plt.gca()
im = ax.imshow(np.arange(200).reshape((20, 10)))
divider = make_axes_locatable(ax)
width = axes_size.AxesY(ax, aspect=1./aspect)
pad = axes_size.Fraction(pad_fraction, width)
cax = divider.append_axes("right", size=width, pad=pad)
plt.colorbar(im, cax=cax)

Perhatikan bahwa ini menentukan lebar colorbar dengan ketinggian plot (berbeda dengan lebar gambar, seperti sebelumnya).

Jarak antara colorbar dan plot sekarang dapat ditentukan sebagai sebagian kecil dari lebar colorbar, yang merupakan angka yang jauh lebih bermakna dari pada sebagian kecil dari lebar gambar.

MEMPERBARUI:

Saya membuat notebook IPython pada topik , di mana saya mengemas kode di atas menjadi fungsi yang dapat digunakan kembali:

import matplotlib.pyplot as plt
from mpl_toolkits import axes_grid1

def add_colorbar(im, aspect=20, pad_fraction=0.5, **kwargs):
    """Add a vertical color bar to an image plot."""
    divider = axes_grid1.make_axes_locatable(im.axes)
    width = axes_grid1.axes_size.AxesY(im.axes, aspect=1./aspect)
    pad = axes_grid1.axes_size.Fraction(pad_fraction, width)
    current_ax = plt.gca()
    cax = divider.append_axes("right", size=width, pad=pad)
    plt.sca(current_ax)
    return im.axes.figure.colorbar(im, cax=cax, **kwargs)

Dapat digunakan seperti ini:

im = plt.imshow(np.arange(200).reshape((20, 10)))
add_colorbar(im)

Saya menghargai semua jawaban di atas. Namun, seperti beberapa jawaban dan komentar mencontohkan, axes_grid1modul tidak bisa GeoAxes alamat, sedangkan menyesuaikan fraction, pad, shrink, dan parameter lain yang serupa dapat belum tentu memberikan perintah yang sangat tepat, yang benar-benar mengganggu saya. Saya percaya bahwa memberikan colorbarsendiri axesmungkin merupakan solusi yang lebih baik untuk mengatasi semua masalah yang telah disebutkan.

Kode

import matplotlib.pyplot as plt
import numpy as np

fig=plt.figure()
ax = plt.axes()
im = ax.imshow(np.arange(100).reshape((10,10)))

# Create an axes for colorbar. The position of the axes is calculated based on the position of ax.
# You can change 0.01 to adjust the distance between the main image and the colorbar.
# You can change 0.02 to adjust the width of the colorbar.
# This practice is universal for both subplots and GeoAxes.

cax = fig.add_axes([ax.get_position().x1+0.01,ax.get_position().y0,0.02,ax.get_position().height])
plt.colorbar(im, cax=cax) # Similar to fig.colorbar(im, cax = cax)

Hasil

Kemudian, saya menemukan matplotlib.pyplot.colorbardokumentasi resmi juga memberikan axopsi, yaitu sumbu yang ada yang akan memberikan ruang untuk colorbar. Karenanya, ini berguna untuk banyak subplot, lihat berikut.

Kode

fig, ax = plt.subplots(2,1,figsize=(12,8)) # Caution, figsize will also influence positions.
im1 = ax[0].imshow(np.arange(100).reshape((10,10)), vmin = -100, vmax =100)
im2 = ax[1].imshow(np.arange(-100,0).reshape((10,10)), vmin = -100, vmax =100)
fig.colorbar(im1, ax=ax)

Hasil

Sekali lagi, Anda juga dapat mencapai efek yang sama dengan menentukan cax, cara yang lebih akurat dari sudut pandang saya.

Kode

fig, ax = plt.subplots(2,1,figsize=(12,8))
im1 = ax[0].imshow(np.arange(100).reshape((10,10)), vmin = -100, vmax =100)
im2 = ax[1].imshow(np.arange(-100,0).reshape((10,10)), vmin = -100, vmax =100)
cax = fig.add_axes([ax[1].get_position().x1-0.25,ax[1].get_position().y0,0.02,ax[0].get_position().y1-ax[1].get_position().y0])
fig.colorbar(im1, cax=cax)

Hasil

Saat Anda membuat colorbarcoba gunakan fraksi dan / atau susutkan parameter.

Dari dokumen:

fraksi 0,15; sebagian kecil dari sumbu asli yang digunakan untuk colorbar

menyusut 1.0; fraksi yang digunakan untuk mengecilkan colorbar

Semua solusi di atas baik, tapi saya suka @ Steve dan @ bejota yang terbaik karena mereka tidak melibatkan panggilan mewah dan bersifat universal.

Secara universal maksud saya yang bekerja dengan semua jenis sumbu termasuk GeoAxes. Misalnya, Anda telah memproyeksikan sumbu untuk pemetaan:

projection = cartopy.crs.UTM(zone='17N')
ax = plt.axes(projection=projection)
im = ax.imshow(np.arange(200).reshape((20, 10)))

panggilan ke

cax = divider.append_axes("right", size=width, pad=pad)

akan gagal dengan: KeyException: map_projection

Oleh karena itu, satu-satunya cara universal menangani ukuran colorbar dengan semua jenis sumbu adalah:

ax.colorbar(im, fraction=0.046, pad=0.04)

Bekerja dengan pecahan dari 0,035 hingga 0,046 untuk mendapatkan ukuran terbaik Anda. Namun, nilai untuk fraksi dan paddig perlu disesuaikan untuk mendapatkan yang paling cocok untuk plot Anda dan akan berbeda tergantung jika orientasi bar warna berada di posisi vertikal atau horizontal.