def main(args: Namespace):
load_teams(args.database)
league = league_register[get_unique_league(args)]
load_league(args.database, league)
seasons = Season.seasons(league)
if args.history:
seasons = seasons[-args.history:]
head_season = seasons[-1]
head_table = LeagueTable(head_season, args.half)
table_map = head_table.group(args.chunks)
predicates = predicate_table[args.analysis]
matrix = np.zeros(shape=(table_map.number_of_chunks(),
len(predicates.functions)),
dtype=np.int32)
for season in seasons:
table = LeagueTable(season, args.half)
table_map = table.group(args.chunks)
fill_matrix(matrix, table_map, table, predicates, args.half)
datum = pd.DataFrame(matrix)
datum.columns = [
Event.name(func, negate=False, short=True)
for func in predicates.functions
]
datum.index = create_chunk_labels(table_map)
fig, ax = plt.subplots(nrows=1,
ncols=1,
figsize=(10, 10),
constrained_layout=True)
heatmap(datum, cmap='coolwarm', linewidth=0.5, annot=True, fmt='d', ax=ax)
ax.set_ylabel('Positions')
ax.set_xlabel(args.analysis.name.capitalize())
sublists = split_into_contiguous_groups(
[season.year for season in seasons])
title = '{} {} Seasons:{}'.format(league.country, league.name,
to_string(sublists))
if args.half != Half.both:
title = '{} ({} half)'.format(title, args.half.name)
fig.suptitle(title, fontweight='bold', fontsize=14)
plt.show(block=args.block)
def compute_chunked_data(seasons: List[Season], func: Callable, negate: bool,
venue: Venue, half: Half, chunk_size: int):
chunk_to_chart = OrderedDict()
for season in seasons:
table = LeagueTable(season, half)
table_map = table.group(chunk_size)
for chunk_id in range(table_map.number_of_chunks()):
if chunk_id not in chunk_to_chart:
chart = DataUnit(Counter(),
seasons,
positions=table_map.get_rows(chunk_id))
chunk_to_chart[chunk_id] = chart
chart = chunk_to_chart[chunk_id]
for row_id in table_map.get_rows(chunk_id):
team = table[row_id].TEAM
count_events(season, team, venue, half, func, negate, chart)
return list(chunk_to_chart.values())
def main(args: Namespace):
load_teams(args.database)
league = league_register[get_unique_league(args)]
load_league(args.database, league)
seasons = Season.seasons(league)
if not seasons:
messages.error_message("No season data found")
if args.history:
seasons = seasons[-args.history:]
if args.team:
(row, ) = extract_picked_team(args.database, args.team, league)
selected_team = Team.inventory[row[0]]
else:
selected_team = None
func = Event.get(get_unique_event(args))
if args.chunks:
data = compute_chunked_data(seasons, func, args.negate, args.venue,
args.half, args.chunks)
nrows = len(data)
if selected_team is not None:
ncols = 3
else:
ncols = 1
fig, axes = plt.subplots(nrows=len(data),
ncols=ncols,
figsize=(20, 13),
squeeze=False,
constrained_layout=True)
x_limit, _ = find_limits(data)
for i, datum in enumerate(data):
ax = axes[i, 0]
plot(ax, datum, x_limit, args.lines)
if selected_team is not None:
golden_season = seasons[-1]
golden_table = LeagueTable(golden_season, args.half)
golden_map = golden_table.group(args.chunks)
chunk_to_seasons = OrderedDict()
for chunk_id in range(golden_map.number_of_chunks()):
if chunk_id not in chunk_to_seasons:
chunk_to_seasons[chunk_id] = []
for season in seasons:
if season != golden_season:
table = LeagueTable(season, args.half)
table_map = table.group(args.chunks)
if selected_team in season.teams():
position = table.team_position(selected_team)
chunk_id = table_map.get_chunk(position)
chunk_to_seasons[chunk_id].append(season)
chunk_to_datum = OrderedDict()
for chunk_id, chunk_seasons in chunk_to_seasons.items():
if chunk_seasons:
datum = DataUnit(Counter(),
chunk_seasons,
team=selected_team,
positions=golden_map.get_rows(chunk_id))
chunk_to_datum[chunk_id] = datum
for season in seasons:
if season == golden_season:
position = golden_table.team_position(selected_team)
datum = DataUnit(Counter(), [golden_season],
team=selected_team,
positions=[position],
highlight=True)
golden_datum = datum
else:
if selected_team in season.teams():
table = LeagueTable(season, args.half)
table_map = table.group(args.chunks)
position = table.team_position(selected_team)
chunk_id = table_map.get_chunk(position)
datum = chunk_to_datum[chunk_id]
count_events(season, selected_team, args.venue, args.half,
func, args.negate, datum)
position = golden_table.team_position(selected_team)
chunk_id = golden_map.get_chunk(position)
ax = axes[chunk_id, 1]
plot(ax, golden_datum, x_limit, args.lines)
used = {(chunk_id, 1)}
for chunk_id, datum in chunk_to_datum.items():
ax = axes[chunk_id, 2]
plot(ax, datum, x_limit, args.lines)
used.add((chunk_id, 2))
for i in range(0, nrows):
for j in range(1, 3):
if (i, j) not in used:
fig.delaxes(axes[i, j])
else:
data = compute_aggregated_data(seasons, selected_team, func,
args.negate, args.venue, args.half)
x_limit, _ = find_limits(data)
display = DisplayGrid(len(data), 2)
fig, axes = plt.subplots(nrows=display.nrows,
ncols=display.ncols,
figsize=(20, 10),
squeeze=False)
for i, datum in enumerate(data):
cell_x, cell_y = display.index(i)
ax = axes[cell_x, cell_y]
plot(ax, datum, x_limit, args.lines)
title = construct_title(league, func, args.negate, args.venue, args.half)
fig.suptitle(title, fontweight='bold', fontsize=14)
plt.show(block=args.block)
def main(args: Namespace):
load_teams(args.database)
league = league_register[get_unique_league(args)]
load_league(args.database, league)
seasons = Season.seasons(league)
if args.history:
seasons = seasons[-args.history:]
event = Event.get(get_unique_event(args))
head_season = seasons[-1]
head_table = LeagueTable(head_season, args.half)
if head_table.played() == 0:
head_season = seasons[-2]
head_table = LeagueTable(head_season, args.half)
nrows = 1
else:
nrows = 2
head_map = head_table.group(args.chunks)
shape = (head_map.number_of_chunks(), head_map.number_of_chunks())
historical_matrix = np.zeros(shape=shape, dtype=np.int32)
historical_seasons = seasons[:-1]
for season in historical_seasons:
table = LeagueTable(season, args.half)
fill_matrix(event, args.symmetry, historical_matrix, head_map, table,
args.half)
display = DisplayGrid(nrows, 1)
fig, axs = plt.subplots(nrows=display.nrows,
ncols=display.ncols,
figsize=(15, 12),
squeeze=False,
constrained_layout=True)
datum = pd.DataFrame(historical_matrix)
sublists = split_into_contiguous_groups(
[season.year for season in historical_seasons])
title = 'Seasons:{}'.format(to_string(sublists))
cell_x, cell_y = display.index(0)
ax = axs[cell_x, cell_y]
populate_axis(head_map, datum, ax, title)
if nrows == 2:
now_matrix = np.zeros(shape=shape, dtype=np.int32)
fill_matrix(event, args.symmetry, now_matrix, head_map, head_table,
args.half)
datum = pd.DataFrame(now_matrix)
title = 'Season:{}'.format(head_season.year)
cell_x, cell_y = display.index(1)
ax = axs[cell_x, cell_y]
populate_axis(head_map, datum, ax, title)
title = '{} {}: {}'.format(league.country, league.name,
Event.name(event, args.negate))
if args.half != Half.both:
title = '{} ({} half)'.format(title, args.half.name)
fig.suptitle(title, fontweight='bold', fontsize=14)
plt.show(block=args.block)