60 lines
1.3 KiB
Python
60 lines
1.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
Build ammonia production.
|
|
"""
|
|
|
|
import pandas as pd
|
|
|
|
country_to_alpha2 = {
|
|
"Austriae": "AT",
|
|
"Bulgaria": "BG",
|
|
"Belgiume": "BE",
|
|
"Croatia": "HR",
|
|
"Czechia": "CZ",
|
|
"Estonia": "EE",
|
|
"Finland": "FI",
|
|
"France": "FR",
|
|
"Germany": "DE",
|
|
"Greece": "GR",
|
|
"Hungarye": "HU",
|
|
"Italye": "IT",
|
|
"Lithuania": "LT",
|
|
"Netherlands": "NL",
|
|
"Norwaye": "NO",
|
|
"Poland": "PL",
|
|
"Romania": "RO",
|
|
"Serbia": "RS",
|
|
"Slovakia": "SK",
|
|
"Spain": "ES",
|
|
"Switzerland": "CH",
|
|
"United Kingdom": "GB",
|
|
}
|
|
|
|
if __name__ == "__main__":
|
|
if "snakemake" not in globals():
|
|
from helper import mock_snakemake
|
|
|
|
snakemake = mock_snakemake("build_ammonia_production")
|
|
|
|
ammonia = pd.read_excel(
|
|
snakemake.input.usgs,
|
|
sheet_name="T12",
|
|
skiprows=5,
|
|
header=0,
|
|
index_col=0,
|
|
skipfooter=19,
|
|
)
|
|
|
|
ammonia.rename(country_to_alpha2, inplace=True)
|
|
|
|
years = [str(i) for i in range(2013, 2018)]
|
|
countries = country_to_alpha2.values()
|
|
ammonia = ammonia.loc[countries, years].astype(float)
|
|
|
|
# convert from ktonN to ktonNH3
|
|
ammonia *= 17 / 14
|
|
|
|
ammonia.index.name = "ktonNH3/a"
|
|
|
|
ammonia.to_csv(snakemake.output.ammonia_production)
|