matplotlib 中的科学记数法颜色条

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()

当我这样做时，我得到以下图像：

但是，我想以科学记数法查看刻度标签...是否有任何一行命令可以做到这一点？ 否则，那里有任何提示吗？ 谢谢！

Answer 1

您可以使用colorbar的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

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

创建您的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()文档。

PS：也许这最适合发表评论。 但我没有足够的声誉。