choropleth_method_2.py

m = folium.Map(location=[48, -102], zoom_start=3)

folium.Choropleth(
    geo_data=usmap_json_with_id,
    name="choropleth",
    data=all_states_census_df,
    columns=["state_str", "Total_Pop_2021"],
    key_on="feature.id",
    fill_color="YlGn",
    fill_opacity=0.7,
    line_opacity=0.2,
    legend_name="Total Population By State, 2021",
).add_to(m)

folium.LayerControl().add_to(m)

# Display the Choropleth
m