Vedant-acharya's picture
Added 3 category files area,funding and population
1b058a8 verified
def true_code():
import numpy as np
import pandas as pd
main_data = pd.read_csv("raw_data/main_data.csv")
main_data['Timestamp'] = pd.to_datetime(main_data['Timestamp'])
states_data = pd.read_csv("raw_data/State_data.csv")
ncap_funding_data = pd.read_csv("raw_data/NCAP_Funding.csv")
ncap_funding_data.replace('-', np.nan, inplace=True)
ncap_funding_data['Amount released during FY 2019-20'] = ncap_funding_data['Amount released during FY 2019-20'].astype('float64')
ncap_funding_data['Amount released during FY 2020-21'] = ncap_funding_data['Amount released during FY 2020-21'].astype('float64')
ncap_funding_data['Amount released during FY 2021-22'] = ncap_funding_data['Amount released during FY 2021-22'].astype('float64')
ncap_funding_data['Utilisation as on June 2022'] = ncap_funding_data['Utilisation as on June 2022'].astype('float64')
state_averages = main_data.groupby('state')[['PM2.5', 'PM10']].mean()
state_averages['combined'] = state_averages['PM2.5'] + state_averages['PM10']
max_state = state_averages['combined'].idxmax()
state_area = states_data.loc[states_data['state'] == max_state, 'area (km2)'].iloc[0]
print(state_area)
true_code()