我的动画图像没有使用 matplotlib imshow 动画改变

Question

我想在我的代码中对每个蒙特卡罗循环进行动画处理。但是，图像没有更新，并且与开始时的图像保持一致。

ground_state 是一个二维数组，在每次调用模拟后都会发生变化。我已经检查过它确实在更新中发生了变化

fig = pyplot.figure()
im = pyplot.imshow(ground_state, animated = True)

def update_fig(*args):
    global ground_state
    simulate(ground_state, 100000, 2.4)
    im.set_data(ground_state)
    return im

ani = animation.FuncAnimation(fig, update_fig, interval = 50)

pyplot.show()

这是模拟功能：-

def simulate(state, n, T):
    """
    Simulates at temperature T for N iterations.
    """
    def acceptance_ratio(del_E):
        """
        Calculates the acceptance ratio.
        """
        # value of beta = 1 / (boltzmann constant * T)
        beta = 1 / T
        if (del_E) < 0:
            return 1
        else:
            return e ** (-beta * (del_E))

    step = 0
    cycle = 0
    # Single flipping
    for dummy in range(n):
        switch = (random.choice(range(len(state))), 
random.choice(range(len(state))))
        neighbours = nearest_neighbours(switch[0], switch[1], len(state))
        change_E = 0
        for pos in neighbours:
            change_E += 2 * J * state[pos[0]][pos[1]] * state[switch[0][switch[1]]
    p = acceptance_ratio(change_E)
    step += 1
    if step == 100 * 100:
        cycle += 1
    rand_num = random.random()
    if rand_num <= p:
        state[switch[0]][switch[1]] *= -1
    return state

Answer 1

我们不知道simulate 函数内部发生了什么，但似乎这就是问题的根源，因为否则，代码运行良好并具有动画效果，

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

def simulate(*args):
    global ground_state
    ground_state = np.random.randn(5,5)

simulate()

fig = plt.figure()
im = plt.imshow(ground_state, animated = True)

def update_fig(*args):
    global ground_state
    simulate(ground_state, 100000, 2.4)
    im.set_data(ground_state)
    return im

ani = animation.FuncAnimation(fig, update_fig, interval = 50)

plt.show()