对我来说,一个接一个地显示数据似乎有点不符合人体工程学。此外,使用动画可能不是最好的解决方案。如果检查第二个数据集后您想返回第一个数据集怎么办?
因此,我将实施一个允许在光谱之间来回切换的解决方案。
以下沙箱示例基于我为类似的图像问题提供了一个解决方案 https://stackoverflow.com/questions/41143782/paging-scrolling-through-set-of-2d-heat-maps-in-matplotlib/41152160#41152160。它使用离散滑块来浏览页面。尽管乍一看似乎有点复杂,但您实际上不必理解PageSlider
类以便使用它。只需查看下面的代码即可__main__
part.
import matplotlib.widgets
import matplotlib.patches
import mpl_toolkits.axes_grid1
class PageSlider(matplotlib.widgets.Slider):
def __init__(self, ax, label, numpages = 10, valinit=0, valfmt='%1d',
closedmin=True, closedmax=True,
dragging=True, **kwargs):
self.facecolor=kwargs.get('facecolor',"w")
self.activecolor = kwargs.pop('activecolor',"b")
self.fontsize = kwargs.pop('fontsize', 10)
self.numpages = numpages
super(PageSlider, self).__init__(ax, label, 0, numpages,
valinit=valinit, valfmt=valfmt, **kwargs)
self.poly.set_visible(False)
self.vline.set_visible(False)
self.pageRects = []
for i in range(numpages):
facecolor = self.activecolor if i==valinit else self.facecolor
r = matplotlib.patches.Rectangle((float(i)/numpages, 0), 1./numpages, 1,
transform=ax.transAxes, facecolor=facecolor)
ax.add_artist(r)
self.pageRects.append(r)
ax.text(float(i)/numpages+0.5/numpages, 0.5, str(i+1),
ha="center", va="center", transform=ax.transAxes,
fontsize=self.fontsize)
self.valtext.set_visible(False)
divider = mpl_toolkits.axes_grid1.make_axes_locatable(ax)
bax = divider.append_axes("right", size="5%", pad=0.05)
fax = divider.append_axes("right", size="5%", pad=0.05)
self.button_back = matplotlib.widgets.Button(bax, label=ur'$\u25C0$',
color=self.facecolor, hovercolor=self.activecolor)
self.button_forward = matplotlib.widgets.Button(fax, label=ur'$\u25B6$',
color=self.facecolor, hovercolor=self.activecolor)
self.button_back.label.set_fontsize(self.fontsize)
self.button_forward.label.set_fontsize(self.fontsize)
self.button_back.on_clicked(self.backward)
self.button_forward.on_clicked(self.forward)
def _update(self, event):
super(PageSlider, self)._update(event)
i = int(self.val)
if i >=self.valmax:
return
self._colorize(i)
def _colorize(self, i):
for j in range(self.numpages):
self.pageRects[j].set_facecolor(self.facecolor)
self.pageRects[i].set_facecolor(self.activecolor)
def forward(self, event):
current_i = int(self.val)
i = current_i+1
if (i < self.valmin) or (i >= self.valmax):
return
self.set_val(i)
self._colorize(i)
def backward(self, event):
current_i = int(self.val)
i = current_i-1
if (i < self.valmin) or (i >= self.valmax):
return
self.set_val(i)
self._colorize(i)
if __name__ == "__main__":
import numpy as np
from matplotlib import pyplot as plt
num_pages = 10
data = np.random.rand(700, num_pages)
spec = np.linspace(-10,10, 700)
fig, ax = plt.subplots()
fig.subplots_adjust(bottom=0.18)
ax.set_ylim([0.,1.6])
line, = ax.plot(spec,data[:,0], color="b")
ax_slider = fig.add_axes([0.1, 0.05, 0.8, 0.04])
slider = PageSlider(ax_slider, 'Page', num_pages, activecolor="orange")
def update(val):
i = int(slider.val)
line.set_ydata(data[:,i])
slider.on_changed(update)
plt.show()
上面的代码正在运行并显示了它的样子。根据您的具体情况,您需要对其进行一些更改。
我尝试相应地调整您的代码,但当然我不能保证它有效。这段代码必须放在下面__main__
部分,PageSlider 必须保持不变。
import numpy as np
from matplotlib import pyplot as plt
dateien = ['./2450ATT0.csv','./2450ATT0-1.csv','./2450ATT0-2.csv']
data_x = []
data_y = []
for datei in dateien: #do not call a variable "file" in python as this is protected
x = np.genfromtxt(datei, usecols =(0), delimiter=';', unpack=True)
x = x/1000000.
y = np.genfromtxt(datei, usecols =(1), delimiter=';', unpack=True, dtype=float)
data_x.append(x)
data_y.append(y)
fig, ax = plt.subplots()
fig.subplots_adjust(bottom=0.18)
ax.set_xlim([2435,2465])
ax.set_xlim([-120,20])
ax.set_xlabel('Frequenz')
ax.set_ylabel('Leistung')
text = ax.text(0.98,0.98, dateien[0], ha="right", va="top")
line, = ax.plot(data_x[0],data_y[0], color="b")
ax_slider = fig.add_axes([0.1, 0.05, 0.8, 0.04])
slider = PageSlider(ax_slider, 'Page', len(dateien), activecolor="orange")
def update(val):
i = int(slider.val)
line.set_data(data_x[i],data_y[i])
text.set_text(dateien[i])
slider.on_changed(update)
plt.show()
Edit:
对于简单的动画,您宁愿使用matplotlib.animation.FuncAnimation http://matplotlib.org/api/animation_api.html代码看起来就是这样的
import numpy as np
from matplotlib import pyplot as plt
dateien = ['./2450ATT0.csv','./2450ATT0-1.csv','./2450ATT0-2.csv']
data_x = []
data_y = []
for datei in dateien: # do not call a variable "file" in python, this is a protected word
x = np.genfromtxt(datei, usecols =(0), delimiter=';', unpack=True)
x = x/1000000.
y = np.genfromtxt(datei, usecols =(1), delimiter=';', unpack=True, dtype=float)
data_x.append(x)
data_y.append(y)
fig, ax = plt.subplots()
fig.subplots_adjust(bottom=0.18)
ax.set_xlim([2435,2465])
ax.set_xlim([-120,20])
ax.set_xlabel('Frequenz')
ax.set_ylabel('Leistung')
line, = ax.plot(data_x[0],data_y[0], color="b")
def update(i):
line.set_data(data_x[i],data_y[i])
ani = matplotlib.animation.FuncAnimation(fig, update,
frames= len(dateien), interval = 200, blit = False, repeat= True)
plt.show()