在matplotlib中設置y軸系數的顏色

Question

這是我的Python代碼：

import matplotlib.pyplot as plt
import numpy as np

# create data
data = np.random.randn(200, 2)
data = data * 1e20
assert(data.ndim == 2)
assert(data.shape[1] == 2)
x = np.arange(data.shape[0])
fix, ax1 = plt.subplots()
ax1.plot(x, data[:, 0], 'b')
for i in ax1.get_yticklabels():
    i.set_color('b')
ax1.set_ylabel('', color='b')
ax2 = ax1.twinx()
ax2.plot(x, data[:, 1], 'r')
for i in ax2.get_yticklabels():
    i.set_color('r')
plt.show()

我正在嘗試在一個繪圖上繪制兩條共享相同x軸的曲線。 我也想為兩個y軸使用不同的顏色。 我得到結果：

問題是，在y軸的左側，頂部的系數為“ 1e20”。 我也希望也將其染成藍色。 我該如何實現？

Answer 1

在matplotlib中，該系數稱為與其相關聯的軸的“偏移文本”。 因此，您可以通過在相應的軸對象上調用get_offset_text()方法來訪問它。 然后，您可以使用各種格式化方法。 在這種情況下，您將需要在每個y軸上調用set_color()方法。 您可以通過在調用plt.show()之前立即添加以下兩行代碼來完成此操作：

ax1.yaxis.get_offset_text().set_color("blue")
ax2.yaxis.get_offset_text().set_color("red")