Анимируйте 2 разных сюжета друг с другом с помощью Matplotlib

Question

ТАК Я анимировал серию кадров с Matplotlib. Теперь для каждого кадра есть точка c, которую я хочу пометить красной точкой. У меня есть индекс точки каждого кадра в списке. Как я могу добавить нанесение точки на анимацию. вот мой код: data_clean - это основные данные, которые я уже оживляю. pred - список, содержащий точки.

fig = plt.figure(figsize=(10,5)) 

ax= fig.add_subplot(111,xlabel='Range bins', ylabel='Doppler bins' )
im = ax.imshow(data_clean[0,0,...], vmin= -60, vmax=0, animated=True)
cbar = fig.colorbar(im)
cbar.set_label('dB Full Scale')
#this should scatter the dot
im4 = plt.scatter(*pred[:,0], color="r")


def animate(i):
    im3.set_array(data_clean[i,0])
    im4.set_array(pred[:,i])
    return im3, im4


anim = animation.FuncAnimation(fig, animate,
                               frames=50, interval=50, blit=False, repeat=True)