matplotlib中的科学记谱法colorbar

Question

我正在尝试使用matplotlib将颜色条添加到我的图像中.当我试图强制用科学记数法书写刻度标签时,问题出现了.如何在颜色条的刻度中强制使用科学记数法(即1x10 ^ 0,2x10 ^ 0,...,1x10 ^ 2等)？

例如,让我们用它的颜色条创建和绘制图像:

import matplotlib as plot
import numpy as np

img = np.random.randn(300,300)
myplot = plt.imshow(img)
plt.colorbar(myplot)
plt.show()

当我这样做时,我得到以下图像:

但是,我希望看到科学记数法中的ticklabels ...是否有任何一行命令来执行此操作？否则,那里有任何提示吗？谢谢!

Answer 1

你可以使用colorbar's format参数:

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

img = np.random.randn(300,300)
myplot = plt.imshow(img)

def fmt(x, pos):
    a, b = '{:.2e}'.format(x).split('e')
    b = int(b)
    return r'${} \times 10^{{{}}}$'.format(a, b)

plt.colorbar(myplot, format=ticker.FuncFormatter(fmt))
plt.show()

Answer 2

您可以指定颜色栏刻度的格式,如下所示:

pl.colorbar(myplot, format='%.0e')

Answer 3

有一种更直接（但不太可定制）的方法可以在ColorBar没有%.0e格式的情况下获得科学记数法。

创建您的ColorBar：

cbar = plt.colorbar()

并调用格式化程序：

cbar.formatter.set_powerlimits((0, 0))

这将使ColorBar使用科学记数法。请参阅下面的示例图以了解ColorBar将如何显示。

可以在此处找到此功能的文档。

Answer 4

看来，cbar.formatter.set_powerlimits((0,0))约瑟夫的回答中仅一人还没有呈现像 $10^3$ 这样的数学格式。

添加cbar.formatter.set_useMathText(True)类似 $10^3$ 的值。

import matplotlib.pyplot as plt
import numpy as np

img = np.random.randn(300,300)*10**5
myplot = plt.imshow(img)
cbar = plt.colorbar(myplot)
cbar.formatter.set_powerlimits((0, 0))

# to get 10^3 instead of 1e3
cbar.formatter.set_useMathText(True)

plt.show()

这会生成

set_useMathText() 请参阅此处的文档。