【发布时间】：2022-11-28 18:08:50
【问题描述】：
我正在寻找一种更新数组值的方法 通过创建更新函数创建的 numpy 数组 更新前一个数组的值和 更改新值的颜色已更新
below is my code though it only display the final frame.
My Question is how do i display the entire process to show how each cell satisfying the condition given change color step by step.
from tkinter import N
import args as args
import numpy as np
from matplotlib import pyplot as plt, animation
from matplotlib import colors
# setting up the values for the grid
ON = 1
OFF = 0
vals = [ON, OFF]
data = np.array([
[0,0,0,0,1,1,0,0,0,0],
[0,0,0,0,0,0,0,0,0,0],
[0,0,2,0,0,0,0,2,0,0],
[0,0,0,0,0,0,0,0,0,0],
[0,0,0,0,2,0,0,0,0,0],
[0,0,0,0,0,0,0,0,0,0],
[0,0,0,0,0,0,0,0,0,0],
[0,0,0,0,0,0,0,0,0,0]
])
def update(previous_array):
for row in range(previous_array):
for cell in range(row):
if cell > 1:
color = 'red'
return previous_array
cmap = colors.ListedColormap(['Blue','red'])
# plt.figure(figsize=(6,6))
fig, ax = plt.subplots()
# im = ax.imshow(data, cmap=cmap)
plt.pcolor(data[::-1], cmap=cmap, edgecolors='k', linewidths=2)
for i in range(len(data)):
for j in range(len(data[1])):
color = 'red' if data[i, j] == 0 else 'blue'
ani = animation.FuncAnimation(fig, update,
frames=30,
interval=50,
save_count=50)
ani.save('basic_animation.mp4', fps=30, extra_args=['-vcodec', 'libx264'])
plt.show()
``
【问题讨论】：
标签： python arrays animation 2d