diff --git a/violence/plotting.py b/violence/plotting.py index b34cd6e..9a5945d 100644 --- a/violence/plotting.py +++ b/violence/plotting.py @@ -15,7 +15,7 @@ def summary(): try: group_col = re.findall("\['(.*)'\]", file)[0] data = data.groupby(group_col).agg('median').reset_index() - print(data.head()) + print(data[[group_col, 'aggressor_pct']].head(8)) except IndexError: num_steps = int(re.findall("_(.*).", file)[0].split('.')[0].split('_')[-1]) print(f'NĂºmero steps {num_steps}')