动画模拟布朗运动随机游走和停靠效果-Python小屋

照着Python小屋的推送，动手敲了一下动画模拟布朗运动随机游走和停靠效果

from tkinter.messagebox import showinfo
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.animation as animation




#c 表示散点数量
r, c = 2, 30
fig, ax = plt.subplots()
plt.xlim(-40, 40)
plt.ylim(-40, 40)
def init():
    showinfo('hi', 'a new animation - xhc')
    global positions, stop_positions, scatters
    #散点初始位置和预期停靠位置
    positions = np.random.randint(-10, 10, (r, c))
    stop_positions = np.random.randint(-39, 39, (r, c))
    #每个散点的颜色，随机彩色
    colors = np.random.random((c,3))
    scatters = ax.scatter(*positions, marker='*', s = 60, c = colors)
    return scatters,


def update(i):
    global positions
    #随机游走，两个方向随机加减1 ， 都限定在[-39,39]区间内
    offsets = np.random.choice((1,-1),(r,c))
    #已经到达指定坐标的三点不再移动
    offsets[positions==stop_positions] = 0
    if offsets.any():
        positions = positions + offsets
        positions[positions>39] = 39
        positions[positions<-39] = -39
        #更新散点位置
        scatters.set_offsets(positions.T)
        yield scatters
    else:
        init()

ani_scatters = animation.FuncAnimation(fig,init_func=init,func=update,interval=0.5,blit=True)
plt.show()

最后

以上就是幽默雪糕为你收集整理的动画模拟布朗运动随机游走和停靠效果-Python小屋动画模拟布朗运动随机游走和停靠效果-Python小屋的全部内容，希望文章能够帮你解决动画模拟布朗运动随机游走和停靠效果-Python小屋动画模拟布朗运动随机游走和停靠效果-Python小屋所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。