Vedant-acharya's picture
Added 3 category files area,funding and population
1b058a8 verified
def true_code():
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")
state_pm25 = main_data.groupby('state')['PM2.5'].mean().reset_index()
states_area = states_data[['state', 'area (km2)']]
merged_df = state_pm25.merge(states_area, on='state', how='inner')
merged_df['pm25_per_km2'] = merged_df['PM2.5'] / merged_df['area (km2)']
union_territories = ['Delhi', 'Chandigarh', 'Andaman and Nicobar Islands',
'Dadra and Nagar Haveli', 'Daman and Diu', 'Lakshadweep',
'Puducherry', 'Jammu and Kashmir', 'Ladakh']
ut_df = merged_df[merged_df['state'].isin(union_territories)]
lowest_pm25_ut = ut_df.loc[ut_df['pm25_per_km2'].idxmin(), 'state']
print(lowest_pm25_ut)
true_code()