Welcome toVigges Developer Community-Open, Learning,Share
Welcome To Ask or Share your Answers For Others


0 votes
in Technique[技术] by (71.8m points)

python - Pandas function operations

Data is from the United States Census Bureau. Counties are political and geographic subdivisions of states in the United States. This dataset contains population data for counties and states in the US from 2010 to 2015.

Which state has the most counties in it? (hint: consider the sumlevel key carefully! You'll need this for future questions too...)

I can not fetch the county name out of the code. Please help

my code:

import pandas as pd
import numpy as np
census_df = pd.read_csv('census.csv')
def answer_five():
    return census_df.groupby('STNAME').COUNTY.sum().max()

See Question&Answers more detail:os

Welcome To Ask or Share your Answers For Others

1 Answer

0 votes
by (71.8m points)

Here is the answer that worked for me:

def answer_five():
    return census_df.groupby(["STNAME"],sort=False).sum()["COUNTY"].idxmax()

First part created aggregated df


Second part takes the col you need


and returns value corresponding to index with max, check here

Welcome to Vigges Developer Community for programmer and developer-Open, Learning and Share