matplotlib调整colorbar

Question

我有一个2D数据集，其值介于0.5和2之间。 我想用imshow和seismic颜色贴图显示它，但我需要值1来匹配白色。

import numpy as np
import matplotlib.pyplot as plt
data = np.random.random((100,100))*2
data[data<0.5]=0.5
plt.imshow(data,cmap='seismic')
plt.colorbar()

我做了：

plt.imshow(data,cmap='seismic',vmin=0,vmax=2)

但我的颜色栏中没有0-0.5值。

由于

Answer 1

使用ColorBar.set_clim()：

import numpy as np
import matplotlib.pyplot as plt
data = np.random.random((100,100))*2
data[data<0.5]=0.5
plt.imshow(data,cmap='seismic', vmin=0.5, vmax=2)
c = plt.colorbar()
c.set_clim(0, 2)

输出：