30 lines
836 B
Python
30 lines
836 B
Python
# coding: utf-8
|
|
|
|
import logging
|
|
import pandas as pd
|
|
from scipy.spatial import cKDTree as KDTree
|
|
|
|
import pypsa
|
|
import powerplantmatching as ppm
|
|
|
|
if 'snakemake' not in globals():
|
|
from vresutils.snakemake import MockSnakemake, Dict
|
|
|
|
snakemake = MockSnakemake(
|
|
input=Dict(base_network='networks/base.nc'),
|
|
output=['resources/powerplants.csv']
|
|
)
|
|
|
|
logging.basicConfig(level=snakemake.config['logging_level'])
|
|
|
|
n = pypsa.Network(snakemake.input.base_network)
|
|
|
|
ppl = ppm.collection.MATCHED_dataset(include_unavailables=True)
|
|
ppl = ppl.loc[ppl.lon.notnull() & ppl.lat.notnull()]
|
|
|
|
substation_lv_i = n.buses.index[n.buses['substation_lv']]
|
|
kdtree = KDTree(n.buses.loc[substation_lv_i, ['x','y']].values)
|
|
ppl = ppl.assign(bus=substation_lv_i[kdtree.query(ppl[['lon','lat']].values)[1]])
|
|
|
|
ppl.to_csv(snakemake.output[0])
|