def get_pop_stats(agent_stats, pop=None):
    pops = (agent_stats[0].keys() if (pop is None) else [pop])
    stats = [stats_i[p] for p in pops for stats_i in agent_stats]
    if (len(stats[0].shape) == 2):
        return np.vstack(stats)
    elif (len(stats[0].shape) == 1):
        return np.hstack(stats)
    raise Exception("Oy! Dafuk type o' agent data is this?")
def get_pop_stats(agent_stats, pop=None):
    pops = (agent_stats[0].keys() if (pop is None) else [pop])
    stats = [stats_i[p] for p in pops for stats_i in agent_stats]
    if (len(stats[0].shape) == 2):
        return np.vstack(stats)
    elif (len(stats[0].shape) == 1):
        return np.hstack(stats)
    raise Exception("Oy! Dafuk type o' agent data is this?")