capacities = {
    r_id: routes_capacities['capacity'][routes_capacities['route_id'] == r_id].values[0]
    for r_id in routes['id']
}