15

我有一个简单的散点图，其中每个点的颜色由 0 到 1 之间的值给出，并设置为选定的颜色图。这是MWE我的一段代码：

import matplotlib.pyplot as plt 
import numpy as np
import matplotlib.gridspec as gridspec

x = np.random.randn(60) 
y = np.random.randn(60)
z = [np.random.random() for _ in range(60)]

fig = plt.figure()
gs = gridspec.GridSpec(1, 2)

ax0 = plt.subplot(gs[0, 0])
plt.scatter(x, y, s=20)

ax1 = plt.subplot(gs[0, 1])
cm = plt.cm.get_cmap('RdYlBu_r')
plt.scatter(x, y, s=20 ,c=z, cmap=cm)
cbaxes = fig.add_axes([0.6, 0.12, 0.1, 0.02]) 
plt.colorbar(cax=cbaxes, ticks=[0.,1], orientation='horizontal')

fig.tight_layout()
plt.show()

看起来像这样：

图片

这里的问题是我想要情节左下方的小水平颜色条位置，但使用该cax参数不仅感觉有点hacky，它显然与tight_layout导致警告的冲突：

/usr/local/lib/python2.7/dist-packages/matplotlib/figure.py:1533: UserWarning: This figure includes Axes that are not compatible with tight_layout, so its results might be incorrect.
  warnings.warn("This figure includes Axes that are not "

难道没有更好的方法来定位颜色条，即当您运行代码时不会向您抛出令人讨厌的警告？

编辑

我希望颜色条只显示最大值和最小值，即：0 和 1，Joe 通过添加如下内容帮助我做到了这vmin=0, vmax=1一点scatter：

plt.scatter(x, y, s=20, vmin=0, vmax=1)

所以我要删除这部分问题。

4

1 回答 1

20

可以使用 ampl_toolkits.axes_grid1.inset_locator.inset_axes将轴放置在另一个轴内。此轴可用于托管颜色条。它的位置是相对于父轴的，类似于放置图例的方式，使用loc参数（例如，loc=3表示左下角）。它的宽度和高度可以用绝对数字（英寸）或相对于父轴（百分比）来指定。

cbaxes = inset_axes(ax1, width="30%", height="3%", loc=3)

在此处输入图像描述

import matplotlib.pyplot as plt 
import numpy as np
import matplotlib.gridspec as gridspec
from mpl_toolkits.axes_grid1.inset_locator import inset_axes

x = np.random.randn(60) 
y = np.random.randn(60)
z = [np.random.random() for _ in range(60)]

fig = plt.figure()
gs = gridspec.GridSpec(1, 2)

ax0 = plt.subplot(gs[0, 0])
plt.scatter(x, y, s=20)

ax1 = plt.subplot(gs[0, 1])
cm = plt.cm.get_cmap('RdYlBu_r')
plt.scatter(x, y, s=20 ,c=z, cmap=cm)

fig.tight_layout()

cbaxes = inset_axes(ax1, width="30%", height="3%", loc=3) 
plt.colorbar(cax=cbaxes, ticks=[0.,1], orientation='horizontal')


plt.show()

请注意，为了抑制警告，可以tight_layout在添加插入轴之前简单地调用。

于 2017-07-02T08:01:21.693 回答