如何将参数传递给 animation.FuncAnimation()？

Question

如何将参数传递给animation.FuncAnimation()？我试过了，但没用。 animation.FuncAnimation() 的签名是

class matplotlib.animation.FuncAnimation(fig, func, frames=None, init_func=None, fargs=None, save_count=None, **kwargs) Bases: matplotlib.animation.TimedAnimation

我已经在下面粘贴了我的代码。我必须做出哪些改变？

import matplotlib.pyplot as plt
import matplotlib.animation as animation

def animate(i,argu):
    print argu

    graph_data = open('example.txt','r').read()
    lines = graph_data.split('\n')
    xs = []
    ys = []
    for line in lines:
        if len(line) > 1:
            x, y = line.split(',')
            xs.append(x)
            ys.append(y)
        ax1.clear()
        ax1.plot(xs, ys)
        plt.grid()

ani = animation.FuncAnimation(fig,animate,fargs = 5,interval = 100)
plt.show()

Answer 1

我想你已经差不多了，下面有一些小的调整，基本上你需要定义一个图形，使用轴手柄并将 fargs 放在列表中，

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

fig, ax1 = plt.subplots(1,1)

def animate(i,argu):
    print(i, argu)

    #graph_data = open('example.txt','r').read()
    graph_data = "1, 1 \n 2, 4 \n 3, 9 \n 4, 16 \n"
    lines = graph_data.split('\n')
    xs = []
    ys = []
    for line in lines:
        if len(line) > 1:
            x, y = line.split(',')
            xs.append(float(x))
            ys.append(float(y)+np.sin(2.*np.pi*i/10))
        ax1.clear()
        ax1.plot(xs, ys)
        plt.grid()

ani = animation.FuncAnimation(fig, animate, fargs=[5],interval = 100)
plt.show()

我将 example.txt 替换为硬连线字符串，因为我没有该文件并添加了对 i 的依赖，因此情节移动了。

Answer 2

检查这个简单的例子：

# -*- coding: utf-8 -*-
import matplotlib.pyplot as plt 
import matplotlib.animation as animation
import numpy as np

data = np.loadtxt("example.txt", delimiter=",")
x = data[:,0]
y = data[:,1]

fig = plt.figure()
ax = fig.add_subplot(111)
line, = ax.plot([],[], '-')
line2, = ax.plot([],[],'--')
ax.set_xlim(np.min(x), np.max(x))
ax.set_ylim(np.min(y), np.max(y))

def animate(i,factor):
    line.set_xdata(x[:i])
    line.set_ydata(y[:i])
    line2.set_xdata(x[:i])
    line2.set_ydata(factor*y[:i])
    return line,line2

K = 0.75 # any factor 
ani = animation.FuncAnimation(fig, animate, frames=len(x), fargs=(K,),
                              interval=100, blit=True)
plt.show()

首先，对于数据处理推荐使用NumPy，最简单的读写数据。

您不必在每个动画步骤中使用 "plot" 函数，而是使用 set_xdata 和 set_ydata 方法来更新数据。

还回顾了 Matplotlib 文档的示例：http://matplotlib.org/1.4.1/examples/animation/。

Answer 3

简介

您将在下面找到如何将参数正确传递给 animation.funcAnimation 函数的代码示例。

如果您将下面的所有代码部分保存为单个 .py 文件，您可以在终端中按如下方式调用脚本： $python3 scriptLiveUpdateGraph.py -d data.csv 其中 data.csv 是包含您要实时显示的数据的数据文件。

常用模块导入

下面是我的脚本开始：

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.animation as animation
import argparse
import time
import os

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

部分函数

这里我声明了animation.funcAnimation函数稍后调用的函数

def animate(i, pathToMeas):
    pullData = open(pathToMeas,'r').read()
    dataArray = pullData.split('\n')
    xar = []
    yar = []
    colunmNames = dataArray[0].split(',')
    # my data file had this structure:
    #col1, col2
    #100, 500
    #95, 488
    #90, 456
    #...
    # and this data file can be updated when the script is running
    for eachLine in dataArray[1:]:
        if len(eachLine) > 1:
           x, y = eachLine.split(',')
           xar.append(float(x))
           yar.append(float(y))

   # convert list to array
   xar = np.asarray(xar)
   yar = np.asarray(yar)

   # sort the data on the x, I do that for the problem I was trying to solve.
   index_sort_ = np.argsort(xar)
   xar = xar[index_sort_]
   yar = yar[index_sort_]

   ax1.clear()
   ax1.plot(xar, yar,'-+')
   ax1.set_xlim(0,np.max(xar))
   ax1.set_ylim(0,np.max(yar))

处理输入参数

为了使脚本更具交互性，我添加了使用 argparse:

读取输入文件的可能性

parser = argparse.ArgumentParser()
parser.add_argument("-d","--data",
                help="data path to the data to be displayed.",
                type=str)

args = parser.parse_args()

调用函数做动画

知道我们正在回答这个话题的主要问题：

ani = animation.FuncAnimation(fig, animate, fargs=(args.data,), interval=1000 )
plt.show()

如何将参数传递给 animation.FuncAnimation()？

How to pass arguments to animation.FuncAnimation()?

python

wxpython

matplotlib

简介

常用模块导入

部分函数

处理输入参数

调用函数做动画