matplotlib 独立颜色条

fsi0uk1n  于 2023-06-23  发布在  其他
关注(0)|答案(6)|浏览(133)

我正在用matplotlib在python中渲染一些图形,并将它们包含在LaTeX论文中(使用LaTeX的漂亮表格对齐,而不是摆弄matplotlib的ImageGrid等)。我想使用savefig创建并保存独立的颜色条,而不需要使用imshow
(the vlim, vmax参数,以及可以显式提供cmap
我能找到的唯一方法是相当复杂的,并且(据我所知)在画布上绘制一个硬编码的矩形:http://matplotlib.org/examples/api/colorbar_only.html

有没有一种优雅的方法可以用matplotlib创建一个独立的colorbar?

xoshrz7s

xoshrz7s1#

你可以创建一些虚拟图像,然后隐藏它的斧头。在自定义轴中绘制颜色条。

import pylab as pl
import numpy as np

a = np.array([[0,1]])
pl.figure(figsize=(9, 1.5))
img = pl.imshow(a, cmap="Blues")
pl.gca().set_visible(False)
cax = pl.axes([0.1, 0.2, 0.8, 0.6])
pl.colorbar(orientation="h", cax=cax)
pl.savefig("colorbar.pdf")

结果:

m528fe3b

m528fe3b2#

使用与HYRY的答案相同的想法,如果您想要一个“独立”的颜色条,即它独立于图形上的项目(不直接与它们的颜色有关),您可以执行以下操作:

from matplotlib import pyplot as plt
import numpy as np

# create dummy invisible image
# (use the colormap you want to have on the colorbar)
img = plt.imshow(np.array([[0,1]]), cmap="Oranges")
img.set_visible(False)

plt.colorbar(orientation="vertical")

# add any other things you want to the figure.
plt.plot(np.random.rand(30))
91zkwejq

91zkwejq3#

所以,基于this answer here,如果你和我一样,想避免这个丑陋的伪plt.imshow(),你可以用两行代码来完成:

import matplotlib as mpl
import matplotlib.pyplot as plt

fig, ax = plt.subplots()
col_map = plt.get_cmap('nipy_spectral')
mpl.colorbar.ColorbarBase(ax, cmap=col_map, orientation = 'vertical')

# As for a more fancy example, you can also give an axes by hand:
c_map_ax = fig.add_axes([0.2, 0.8, 0.6, 0.02])
c_map_ax.axes.get_xaxis().set_visible(False)
c_map_ax.axes.get_yaxis().set_visible(False)

# and create another colorbar with:
mpl.colorbar.ColorbarBase(c_map_ax, cmap=col_map, orientation = 'horizontal')

nhjlsmyf

nhjlsmyf4#

http://matplotlib.org/examples/api/colorbar_only.html的引用为我解决了这个问题。那个例子有点冗长,所以这里有一个简单的方法来制作一个独立的颜色条(为了子孙后代)...

import matplotlib.pyplot as plt
import matplotlib as mpl

fig = plt.figure()
ax = fig.add_axes([0.05, 0.80, 0.9, 0.1])

cb = mpl.colorbar.ColorbarBase(ax, orientation='horizontal', 
                               cmap='RdBu')

plt.savefig('just_colorbar', bbox_inches='tight')

当然,您可以指定Colorbar的许多其他方面

import matplotlib.pyplot as plt
import matplotlib as mpl

fig = plt.figure()
ax = fig.add_axes([0.05, 0.80, 0.9, 0.1])

cb = mpl.colorbar.ColorbarBase(ax, orientation='horizontal', 
                               cmap='gist_ncar',
                               norm=mpl.colors.Normalize(0, 10),  # vmax and vmin
                               extend='both',
                               label='This is a label',
                               ticks=[0, 3, 6, 9])

plt.savefig('just_colorbar', bbox_inches='tight')

wqsoz72f

wqsoz72f5#

该方法也可用于绘制与ax的内容无关的颜色条。设置fraction = .05即可。

编码

import matplotlib as mpl
import matplotlib.pyplot as plt

fig, ax = plt.subplots(1, 1)

fraction = 1  # .05

norm = mpl.colors.Normalize(vmin=-3, vmax=99)
cbar = ax.figure.colorbar(
            mpl.cm.ScalarMappable(norm=norm, cmap='Blues'),
            ax=ax, pad=.05, extend='both', fraction=fraction)

ax.axis('off')
plt.show()

pkmbmrz7

pkmbmrz76#

为了补充@blaylockbk的答案(这是一个很好的解决方案,甚至是添加到已经创建的图),对我来说,cmap参数不会接受字符串来表示颜色Map名称,但是cmap = plt.cm.viridis工作,如果有人遇到和我一样的问题。

相关问题