Pyplot - 限制 x 轴后重新缩放 y 轴

Question

我正在尝试使用 pyplot 绘制一些数据，然后使用 xlim() x 轴“放大”。然而，当我这样做时，新图不会重新调整 y 轴 - 我做错了什么吗？

示例 - 在此代码中，绘图 y 轴范围仍取最大值 20，而不是 10。：

from pylab import *

x = range(20)
y = range(20)

xlim(0,10)
autoscale(enable=True, axis='y', tight=None)
scatter(x,y)
show()
close()

Answer 1

意识到这是一个古老的问题，但这就是我（混乱地）解决这个问题的方法：

1. 使用 .plot() 而不是 .scatter()
2. 访问绘图数据稍后（即使在某处返回图形之后）使用 ax.get_lines()[0].get_xydata()
3. 使用该数据将 y 轴重新缩放为 xlims

片段应该可以工作：

import matplotlib.pyplot as plt
import numpy as np

fig = plt.figure()
ax = fig.add_subplot(1, 1, 1)

x = range(20)
y = range(20)

xlims = [0, 10]

ax.set_xlim(xlims)
ax.plot(x, y, marker='.', ls='')

# pull plot data
data = ax.get_lines()[0].get_xydata()

# cut out data in xlims window
data = data[np.logical_and(data[:, 0] >= xlims[0], data[:, 0] <= xlims[1])]

# rescale y
ax.set_ylim(np.min(data[:, 1]), np.max(data[:, 1]))

plt.show()

Answer 2

我不知道，尽管你可以尝试手动过滤点

scatter([(a,b) for a,b in zip(x,y) if a>0 and a<10])