绘图上不需要的调色板

时间:2018-11-27 15:07:38

标签: python matplotlib colorbar color-palette

我在创建的图上遇到了问题,该图上出现不必要的其他调色板。

我的脚本用于列出数据以创建带有彩色点的图。

plt.close('all')
fig, axes = plt.subplots(nrows = 1, ncols = 1)
fig.set_facecolor('white')
axes.set_ylabel('$dz$ [$\AA$]')
axes.set_xlabel('Time [ns]')
axes.spines['right'].set_visible(False)
axes.spines['top'].set_visible(False)
axes.yaxis.set_ticks_position('left')
axes.xaxis.set_ticks_position('bottom')
axes.tick_params(direction='out')
#axes.set_title('N/A')
axes.set_ylim(-20,10)
axes.set_xlim(0, 90)
cmap = plt.get_cmap('plasma')
colors = [cmap(i) for i in np.linspace(0, 1, 9)]
# Make Color Bar ------------------------------------------------------
cax = 0
divider = make_axes_locatable(axes)
cax = divider.append_axes('right', size='5%', pad=0.1)
im = axes.imshow(np.linspace(1, 8.5, 100).reshape(10, 10), cmap='plasma')
fig.colorbar(im, cax=cax)
#----------------------------------------------------------------------
for i, dist in enumerate(dz):
    if i % 100 == 0:
        x = i / 1000 
        y = dist
        phval = final_pH_array[i]
        axes.plot(x, y, 'k.', markersize = 4 , color = colors[int(phval)], clip_on = False)
plt.savefig('plot.pdf')
plt.show()

结果如下:

enter image description here

如您所见,我不想在绘图中使用其他颜色条/调色板,但似乎无法摆脱它。

任何对此的帮助将是巨大的!

1 个答案:

答案 0 :(得分:1)

我认为im.set_visible(False)应该达到您想要的目标。

但是也许您应该看看plt.scatterscatter返回一个PathCollection,您可以将其传递给colorbar函数。

相关问题