如何定义matplotlib FuncAnimation的func参数

Question

经过多次尝试，我不知道func参数是如何为条形图设置动画的。我有以下代码：

# Imports
import random
from tkinter import *
import matplotlib.pyplot as plt
import matplotlib.animation as animation
import pandas as pd
# Functions
def gen_rand_array(no=500):
    return_list = []
    for x in range(no):
        return_list.append(random.randint(1,101))
    return return_list
def bubblesort(list):
    for i in range(len(list)):
        for j in range(0,len(list) - i - 1):
            if list[j] > list[j+1]:
                list[j],list[j+1] = list[j+1],list[j]
    yield list
# Runtime
def main():
    # User parameters
    win = Tk()
    win.title("Set Parameters")
    win.minsize(500,500)
    array_size = Scale(win,from_=2,to=5000,orient="horizontal",label="Use the slider to set the size of the initial array:",length=450,sliderlength=10).pack()
    win.mainloop()
    # Build random unsorted list
    unsorted_vals = []
    for index,value in enumerate(gen_rand_array()):
        unsorted_vals.append(value)
    # Formatting
    fig,ax = plt.subplots(figsize=(15,8))
    def animate(x):
        ax.set_title("Test")
        ax.set_ylim(0,100)
        for key,spine in ax.spines.items():
            spine.set_visible(False)
        ax.tick_params(bottom="off",left="off",right="off",top="off")
        ax.set_yticks([])
        ax.set_xticks([])
    ax.bar(range(len(unsorted_vals)), unsorted_vals)
    # Visualise the sort
    sorter = bubblesort(unsorted_vals)
    anim = animation.FuncAnimation(fig,frames=sorter,func=animate)
    plt.show()
main()

因此，对于bubblesort((的每次迭代，我都想在条形图上按顺序设置变化的动画。我已经将框架定义为我的生成器，但我不明白需要传递给"func"参数的内容。我一直在网上查找例子，但我仍然不明白我需要在这里放什么。每当我运行代码时，我都会遇到无法解释问题的错误。

Answer 1

在animate()内部，您需要获得部分排序列表的下一个状态，并且需要相应地更新条形图。

通过sorter迭代得到更新的列表，但是根据放置yield语句的嵌套for loops的哪一层，更新的频率或多或少会有所不同。目前，您只生成一次排序列表。我把它向下移动了一个循环，但你可以更频繁地更新：

绘图的更新通过h_bar执行(也可参见此问题以更新条形图(

我省略了代码的某些部分，以集中精力处理主要内容。

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.animation as animation
def bubblesort(list):
    for i in range(len(list)):
        for j in range(0, len(list) - i - 1):
            if list[j] > list[j+1]:
                list[j], list[j+1] = list[j+1], list[j]
                 # yield list  # two elements swapped place
        yield list  # ith bubble loop completed 
    # yield list  # sorted list
n = 100
unsorted_vals = np.random.randint(0, 101, size=n)
sorter = bubblesort(unsorted_vals)
fig, ax = plt.subplots(figsize=(15, 8))
ax.set_ylim(0, 100)
for key, spine in ax.spines.items():
    spine.set_visible(False)
ax.set_yticks([])
ax.set_xticks([])
h_bar = ax.bar(range(len(unsorted_vals)), unsorted_vals)
def animate(x):
    current_list = sorter.__next__()
    for rect, h in zip(h_bar, current_list):
        rect.set_height(h)
anim = animation.FuncAnimation(fig, frames=n, func=animate, repeat=False)
plt.show()

现在你总是更新整个条形图，当只需要更新切换位置的两个元素时，也许你可以通过以下方式加快动画速度：

def bubblesort(list):
    for i in range(len(list)):
        for j in range(0, len(list) - i - 1):
            if list[j] > list[j+1]:
                list[j], list[j+1] = list[j+1], list[j]
                yield list, j
def animate(x):
    current_list, idx = sorter.__next__()
    h_bar[idx].set_height(current_list[idx])
    h_bar[idx+1].set_height(current_list[idx+1])
    return h_bar