alpha=1,
zorder=20))
if display_num:
plt.text(dfFrame.loc[pid]['x']-1,dfFrame.loc[pid]['y']-1.3,str(pid),fontsize=8, color='black', zorder=30)
return fig, ax, dfFrame
anim = VideoClip(lambda x: mplfig_to_npimage(draw_frame(x)[0]), duration=length)
from scipy.spatial import Voronoi, voronoi_plot_2d
vor = Voronoi(anim)
import matplotlib.pyplot as plt
voronoi_plot_2d(anim)
plt.show()
#to save the animation to a file, uncomment the next line
anim.to_videofile(r'C:\Users\Home\Documents\working with positional data - version 1.mp4', fps=fps)
def count_players(dfFrame, pid):
count = dfFrame.join(dfPlayers.team)[dfFrame['x']<=dfFrame.loc[pid]['x']].groupby('team').agg('count').max(axis=1)
try:
num_attack = count['attack']
except KeyError:
num_attack = 0
try:
num_defense = count['defense']
except KeyError:
num_defense = 0
return (num_attack-num_defense)
def draw_area(t):
fig, ax, dfFrame = draw_frame(t)
def __init__(self):
self.fig, self.ax = plt.subplots(1, 1)
anim = VideoClip(lambda x: self.draw_swarmplot(x, self.fig, self.ax), duration=2)
anim.to_videofile('swarmplot.mp4', fps=fps)
