matplotlib的散点图改变了另一个图的轴

Question

我想要合并一个contourf()绘图和一个scatter()绘图matplotlib.pyplot,但是,添加散点图会改变绘图的轴.这是一个小例子:

from matplotlib import pyplot as plt
import numpy as np

data = np.random.rand(10,10)
plt.contourf(data)
# plt.scatter(3, 7, s=200, color='k')
plt.show()

这个脚本创建了一些像这样的情节:

取消注释plt.scatter(3, 7, s=200, color='k')轴限制时,会在轮廓图周围产生白框.可能是使用plt.xlim和手动设置轴限制plt.ylim,但这似乎是一个不必要的麻烦.我怎样才能优雅地完成这项工作？

Answer 1

加

plt.xlim(auto=False)
plt.ylim(auto=False)

countorf一条又一条; 或者你可能总是这样做

xl, yl = plt.xlim(), plt.ylim()

代码中的任何地方,您都对x,y限制感到满意,然后最终做到了

plt.xlim(xl)
plt.ylim(yl)