簡體   English   中英

在matplotlib中向colorbar添加標記或線條

[英]Adding markers or lines to colorbar in matplotlib

我有以下幾行代碼來生成熱圖( pcolormesh )。

import matplotlib.pyplot as plt
import numpy as np

vals = np.linspace(-np.pi/2, np.pi/2, 101)
x, y = np.meshgrid(vals, vals)

z = np.abs(np.sinc(x) * np.sinc(y))

xDeg = np.rad2deg(x)
yDeg = np.rad2deg(y)

plt.pcolormesh(xDeg, yDeg, z, cmap = 'jet', vmin = 0, vmax = 1)
plt.colorbar()

plt.axis([-90, 90, -90, 90])
ticks = np.linspace(-90, 90, 13)
plt.xticks(ticks)
plt.yticks(ticks)

print np.mean(z)                # 0.186225110029
print np.sqrt(np.mean(z**2))    # 0.295710882276
plt.show()

它產生這個圖像: pcolormesh結果

我想放置:

  1. 平均顏色條上的標記(0.186)。
  2. 色條上的水平線,RMS值(0.295)。

有什么輸入?

這比預期的要容易。 我沒有意識到colorbar在plt.colorbar().ax有一個可繪制的軸

cb = plt.colorbar()
cb.ax.plot(0.5, mean, 'w.') # my data is between 0 and 1
cb.ax.plot([0, 1], [rms]*2, 'w') # my data is between 0 and 1
# Note: would need to scale mean/rms to be between 0 and 1

看起來像x軸是(0,1)和y軸 (dataMin,dataMax) (0,1)。

pcolormesh與colorbar上的標記。

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM