Python、Matplotlib、subplot:如何设置轴范围?

Posted

技术标签:

【中文标题】Python、Matplotlib、subplot:如何设置轴范围?【英文标题】:Python, Matplotlib, subplot: How to set the axis range? 【发布时间】:2011-02-20 09:56:06 【问题描述】:

如何将第二个子图的 y 轴范围设置为例如[0,1000] ? 我的数据(文本文件中的一列)的 FFT 图导致(inf.?)尖峰,因此实际数据不可见。

pylab.ylim([0,1000])

不幸的是,没有效果。这是整个脚本:

# based on http://www.swharden.com/blog/2009-01-21-signal-filtering-with-python/
import numpy, scipy, pylab, random

xs = []
rawsignal = []
with open("test.dat", 'r') as f:
      for line in f:
            if line[0] != '#' and len(line) > 0:
                xs.append( int( line.split()[0] ) )
                rawsignal.append( int( line.split()[1] ) )

h, w = 3, 1
pylab.figure(figsize=(12,9))
pylab.subplots_adjust(hspace=.7)

pylab.subplot(h,w,1)
pylab.title("Signal")
pylab.plot(xs,rawsignal)

pylab.subplot(h,w,2)
pylab.title("FFT")
fft = scipy.fft(rawsignal)
#~ pylab.axis([None,None,0,1000])
pylab.ylim([0,1000])
pylab.plot(abs(fft))

pylab.savefig("SIG.png",dpi=200)
pylab.show()

也感谢其他改进!

【问题讨论】:

另见***.com/questions/15858192/… 这能回答你的问题吗? setting y-axis limit in matplotlib 【参考方案1】:

你有pylab.ylim:

pylab.ylim([0,1000])

注意:该命令必须在剧情之后执行!

2021 年更新 因为the use of pylab is now strongly discouraged by matplotlib,你应该改用pyplot:

from matplotlib import pyplot as plt
plt.ylim(0, 100) 
#corresponding function for the x-axis
plt.xlim(1, 1000)

【讨论】:

每当我这样做时,它都会将图像翻转过来。 如果我将它与 hexbin 一起使用,在 plot() 在两个图上都显示白色背景后使用 ylim 如果您不使用绘图,而是使用 savefig 怎么办?? 致电plot(),然后是ylim(),然后是savefig() 在子图中是set_ylim()【参考方案2】:

有时您真的想在绘制数据之前设置坐标轴范围。在这种情况下,您可以设置AxesAxesSubplot 对象的“自动缩放”功能。感兴趣的函数是set_autoscale_onset_autoscalex_onset_autoscaley_on

在您的情况下,您希望冻结 y 轴的限制,但允许 x 轴扩展以容纳您的数据。因此,您要将autoscaley_on 属性更改为False。这是您代码中 FFT 子图 sn-p 的修改版本:

fft_axes = pylab.subplot(h,w,2)
pylab.title("FFT")
fft = scipy.fft(rawsignal)
pylab.ylim([0,1000])
fft_axes.set_autoscaley_on(False)
pylab.plot(abs(fft))

【讨论】:

【参考方案3】:

使用axes objects 是一个很好的方法。如果您想与多个图形和子图进行交互,它会有所帮助。直接添加和操作坐标区对象:

import matplotlib.pyplot as plt
fig = plt.figure(figsize=(12,9))

signal_axes = fig.add_subplot(211)
signal_axes.plot(xs,rawsignal)

fft_axes = fig.add_subplot(212)
fft_axes.set_title("FFT")
fft_axes.set_autoscaley_on(False)
fft_axes.set_ylim([0,1000])
fft = scipy.fft(rawsignal)
fft_axes.plot(abs(fft))

plt.show()

【讨论】:

正如 Rob 所建议的,matplotlib 中的 OO 接口优于基于状态的 pylab 接口。 “虽然很多例子都使用pylab,但不再推荐。对于非交互式绘图,建议使用pyplot创建图形,然后使用OO接口进行绘图。” matplotlib.org/faq/…【参考方案4】:

如果你知道你想要的确切轴,那么

pylab.ylim([0,1000])

按照前面的回答工作。但是,如果您想要一个更灵活的轴来适应您的确切数据,就像我在发现这个问题时所做的那样,那么将轴限制设置为数据集的长度。如果您的数据集是问题中的fft,则在您的绘图命令后添加:

length = (len(fft)) pylab.ylim([0,length])

【讨论】:

【参考方案5】:

如果您有多个子图,即

fig, ax = plt.subplots(4, 2)

您可以对所有这些使用相同的 y 限制。它从第一个图中得到 y 轴的限制。

plt.setp(ax, ylim=ax[0,0].get_ylim())

【讨论】:

以上是关于Python、Matplotlib、subplot:如何设置轴范围?的主要内容,如果未能解决你的问题,请参考以下文章

python使用matplotlib:subplot绘制多个子图

python matplotlib中axes与subplot

python matplotlib subplot 上面面积大下面小怎么办

Python matplotlib subplot 简单使用

python使用matplotlib可视化subplots子图为subplots添加主标题(main title)为subplots的每一个子图添加子图标题(subplots title)

为啥很多例子在 Matplotlib/pyplot/python 中使用 `fig, ax = plt.subplots()`