将 Pandas 数据帧中上层聚合的值连接到下层聚合

Question

I have two Pandas data frames.

The first data frame ( county ) has county-level data -

COUNTY_FIPS    COUNTY_INCOME    COUNTY_PERCENT_UNINSURED
      51001            42260                        16.7
      51003            72265                         7.6

The second data frame ( tract ) has Census tract-level data -

 TRACT_FIPS    TRACT_INCOME    TRACT_PERCENT_UNINSURED
51001090100           48861                       13.4
51001090200           42663                        9.4
51003090300           32532                       19.7
51003090100           55678                       12.1

I would like to join values from upper-level aggregates (county-level data) to the lower-level aggregates (Census tract-level data). Note that the first five numbers of the TRACT_FIPS correspond to which county those Census tracts are in (see COUNTY_FIPS). My final data frame would look like this -

 TRACT_FIPS    TRACT_INCOME    TRACT_PERCENT_UNINSURED    COUNTY_INCOME    COUNTY_PERCENT_UNINSURED
51001090100           48861                       13.4            42260                        16.7
51001090200           42663                        9.4            42260                        16.7 
51003090300           32532                       19.7            72265                         7.6
51003090100           55678                       12.1            72265                         7.6

Here's what I have programmed so far (with some pseudocode) -

county_income_values = [] # empty list of county income values
county_percent_uninsured_values # empty list of county percent uninsured values

for tract_fips in tract['tract_fips']: # iterate through all the tract_fips in the tract_fips column
    for county_fips in county['county_fips']: # iterate through all the county_fips in the county_fips column
        if tract_fips[0:5] == county_fips: # if the first 5 digits of the tract_id match the county_id
            # TO DO: Find the index of where the if statement evaluates to true, and append the 
                     county income value at that index to county_income_values_list
            # TO DO: Find the index of where the if statement evaluates to true, and append the 
                     county percent uninsured value at that index to county_percent_uninsured_values 

If there is a more efficient way to go about solving this problem, then feel free to ignore my code above.

Thanks very much in advance!

Answer 1

You can use the function merge . First, you need to extract the first five digits from the column 'TRACT_FIPS' in the second dataframe. Then you can can convert the column 'COUNTY_FIPS' to string and use both columns to merge on:

left = df2['TRACT_FIPS'].astype('str').str[:5]
right = df1['COUNTY_FIPS'].astype('str')

df2.merge(df1, left_on=left, right_on=right)