|
from csv import DictReader
|
|
import pandas as pd
|
|
|
|
PATH = "src/data/electricity_mix.csv"
|
|
|
|
COUNTRY_CODES = [
|
|
("🌎 World", "WOR"),
|
|
("🇪🇺 Europe", "EEE"),
|
|
("🇿🇼 Zimbabwe", "ZWE"),
|
|
("🇿🇲 Zambia", "ZMB"),
|
|
("🇿🇦 South Africa", "ZAF"),
|
|
("🇾🇪 Yemen", "YEM"),
|
|
("🇻🇳 Vietnam", "VNM"),
|
|
("🇻🇪 Venezuela", "VEN"),
|
|
("🇺🇿 Uzbekistan", "UZB"),
|
|
("🇺🇾 Uruguay", "URY"),
|
|
("🇺🇸 United States", "USA"),
|
|
("🇺🇦 Ukraine", "UKR"),
|
|
("🇹🇿 Tanzania", "TZA"),
|
|
("🇹🇼 Taiwan", "TWN"),
|
|
("🇹🇹 Trinidad and Tobago", "TTO"),
|
|
("🇹🇷 Turkey", "TUR"),
|
|
("🇹🇳 Tunisia", "TUN"),
|
|
("🇹🇲 Turkmenistan", "TKM"),
|
|
("🇹🇯 Tajikistan", "TJK"),
|
|
("🇹🇭 Thailand", "THA"),
|
|
("🇹🇬 Togo", "TGO"),
|
|
("🇸🇾 Syrian Arab Republic", "SYR"),
|
|
("🇸🇻 El Salvador", "SLV"),
|
|
("🇸🇳 Senegal", "SEN"),
|
|
("🇸🇰 Slovak Republic", "SVK"),
|
|
("🇸🇮 Slovenia", "SVN"),
|
|
("🇸🇬 Singapore", "SGP"),
|
|
("🇸🇪 Sweden", "SWE"),
|
|
("🇸🇩 Sudan", "SDN"),
|
|
("🇸🇦 Saudi Arabia", "SAU"),
|
|
("🇷🇺 Russian Federation", "RUS"),
|
|
("🇷🇸 Serbia and Montenegro", "SCG"),
|
|
("🇷🇴 Romania", "ROU"),
|
|
("🇶🇦 Qatar", "QAT"),
|
|
("🇵🇾 Paraguay", "PRY"),
|
|
("🇵🇹 Portugal", "PRT"),
|
|
("🇵🇱 Poland", "POL"),
|
|
("🇵🇰 Pakistan", "PAK"),
|
|
("🇵🇭 Philippines", "PHL"),
|
|
("🇵🇪 Peru", "PER"),
|
|
("🇵🇦 Panama", "PAN"),
|
|
("🇴🇲 Oman", "OMN"),
|
|
("🇳🇿 New Zealand", "NZL"),
|
|
("🇳🇵 Nepal", "NPL"),
|
|
("🇳🇴 Norway", "NOR"),
|
|
("🇳🇱 Netherlands", "NLD"),
|
|
("🇳🇮 Nicaragua", "NIC"),
|
|
("🇳🇬 Nigeria", "NGA"),
|
|
("🇳🇦 Namibia", "NAM"),
|
|
("🇲🇿 Mozambique", "MOZ"),
|
|
("🇲🇾 Malaysia", "MYS"),
|
|
("🇲🇽 Mexico", "MEX"),
|
|
("🇲🇹 Malta", "MLT"),
|
|
("🇲🇳 Mongolia", "MNG"),
|
|
("🇲🇲 Myanmar", "MMR"),
|
|
("🇲🇰 North Macedonia", "MKD"),
|
|
("🇲🇩 Moldova", "MDA"),
|
|
("🇲🇦 Morocco", "MAR"),
|
|
("🇱🇾 Libya", "LBY"),
|
|
("🇱🇻 Latvia", "LVA"),
|
|
("🇱🇺 Luxembourg", "LUX"),
|
|
("🇱🇹 Lithuania", "LTU"),
|
|
("🇱🇰 Sri Lanka", "LKA"),
|
|
("🇱🇧 Lebanon", "LBN"),
|
|
("🇰🇿 Kazakhstan", "KAZ"),
|
|
("🇰🇼 Kuwait", "KWT"),
|
|
("🇰🇷 South Korea", "KOR"),
|
|
("🇰🇵 North Korea", "PRK"),
|
|
("🇰🇭 Cambodia", "KHM"),
|
|
("🇰🇬 Kyrgyz Republic", "KGZ"),
|
|
("🇰🇪 Kenya", "KEN"),
|
|
("🇯🇵 Japan", "JPN"),
|
|
("🇯🇴 Jordan", "JOR"),
|
|
("🇯🇲 Jamaica", "JAM"),
|
|
("🇮🇹 Italy", "ITA"),
|
|
("🇮🇸 Iceland", "ISL"),
|
|
("🇮🇷 Iran", "IRN"),
|
|
("🇮🇶 Iraq", "IRQ"),
|
|
("🇮🇳 India", "IND"),
|
|
("🇮🇱 Israel", "ISR"),
|
|
("🇮🇪 Ireland", "IRL"),
|
|
("🇮🇩 Indonesia", "IDN"),
|
|
("🇭🇺 Hungary", "HUN"),
|
|
("🇭🇹 Haiti", "HTI"),
|
|
("🇭🇷 Croatia", "HRV"),
|
|
("🇭🇳 Honduras", "HND"),
|
|
("🇭🇰 Hong Kong", "HKG"),
|
|
("🇬🇹 Guatemala", "GTM"),
|
|
("🇬🇷 Greece", "GRC"),
|
|
("🇬🇮 Gibraltar", "GIB"),
|
|
("🇬🇭 Ghana", "GHA"),
|
|
("🇬🇪 Georgia", "GEO"),
|
|
("🇬🇧 United Kingdom", "GBR"),
|
|
("🇬🇦 Gabon", "GAB"),
|
|
("🇫🇷 France", "FRA"),
|
|
("🇫🇮 Finland", "FIN"),
|
|
("🇪🇹 Ethiopia", "ETH"),
|
|
("🇪🇸 Spain", "ESP"),
|
|
("🇪🇷 Eritrea", "ERI"),
|
|
("🇪🇬 Egypt", "EGY"),
|
|
("🇪🇪 Estonia", "EST"),
|
|
("🇪🇨 Ecuador", "ECU"),
|
|
("🇩🇿 Algeria", "DZA"),
|
|
("🇩🇴 Dominican Republic", "DOM"),
|
|
("🇩🇰 Denmark", "DNK"),
|
|
("🇩🇪 Germany", "DEU"),
|
|
("🇨🇿 Czech Republic", "CZE"),
|
|
("🇨🇾 Cyprus", "CYP"),
|
|
("🇨🇺 Cuba", "CUB"),
|
|
("🇨🇷 Costa Rica", "CRI"),
|
|
("🇨🇴 Colombia", "COL"),
|
|
("🇨🇳 China", "CHN"),
|
|
("🇨🇲 Cameroon", "CMR"),
|
|
("🇨🇱 Chile", "CHL"),
|
|
("🇨🇮 Cote d'Ivoire", "CIV"),
|
|
("🇨🇭 Switzerland", "CHE"),
|
|
("🇨🇬 Congo", "COG"),
|
|
("🇨🇩 Democratic Republic of the Congo", "COD"),
|
|
("🇨🇦 Canada", "CAN"),
|
|
("🇧🇾 Belarus", "BLR"),
|
|
("🇧🇼 Botswana", "BWA"),
|
|
("🇧🇷 Brazil", "BRA"),
|
|
("🇧🇴 Bolivia", "BOL"),
|
|
("🇧🇳 Brunei", "BRN"),
|
|
("🇧🇯 Benin", "BEN"),
|
|
("🇧🇭 Bahrain", "BHR"),
|
|
("🇧🇬 Bulgaria", "BGR"),
|
|
("🇧🇪 Belgium", "BEL"),
|
|
("🇧🇩 Bangladesh", "BGD"),
|
|
("🇧🇦 Bosnia and Herzegovina", "BIH"),
|
|
("🇦🇿 Azerbaijan", "AZE"),
|
|
("🇦🇺 Australia", "AUS"),
|
|
("🇦🇹 Austria", "AUT"),
|
|
("🇦🇷 Argentina", "ARG"),
|
|
("🇦🇴 Angola", "AGO"),
|
|
("🇦 Netherlands Antilles", "ANT"),
|
|
("🇦🇲 Armenia", "ARM"),
|
|
("🇦🇱 Albania", "ALB"),
|
|
("🇦🇪 United Arab Emirates", "ARE")
|
|
]
|
|
|
|
|
|
def find_electricity_mix(code: str):
|
|
|
|
|
|
res = []
|
|
with open(PATH) as fd:
|
|
csv = DictReader(fd)
|
|
for row in csv:
|
|
res += [float(row[code])]
|
|
return res
|
|
|
|
def dataframe_electricity_mix(countries: list):
|
|
|
|
df = pd.read_csv('src/data/electricity_mix.csv')
|
|
df['name_unit'] = df['name'] + ' (' + df['unit'] + ')'
|
|
df = df[['name_unit'] + [x[1] for x in COUNTRY_CODES if x[0] in countries]]
|
|
|
|
df_melted = df.melt(
|
|
id_vars=['name_unit'],
|
|
value_vars=[x[1] for x in COUNTRY_CODES if x[0] in countries],
|
|
var_name='country',
|
|
value_name='value')
|
|
|
|
df = df_melted.pivot(columns='name_unit',
|
|
index='country',
|
|
values='value')
|
|
|
|
return df |
