Python 如果满足某些条件，则用另一个数据帧列中的值填充空数据帧或数组

您可以:

test_mob_df = pd.DataFrame(
    {
        "geoid_o": [10002, 18039, 18039, 18182, 10006, 18111, 18005, 17001],
        "geoid_d": [10005, 18039, 18111, 18182, 18005, 17004, 18050, 15001],
        "pop_flows": [20, 10, 9, 15, 2, 1, 6, 30],
    }
)

state_county_fip = [18182, 18111, 18005, 18039, 18050, 18001]

out = pd.crosstab(
    test_mob_df.loc[test_mob_df["geoid_o"].isin(state_county_fip), "geoid_o"],
    test_mob_df.loc[test_mob_df["geoid_d"].isin(state_county_fip), "geoid_d"],
    values=test_mob_df["pop_flows"],
    aggfunc="first",
)

out = (
    out.reindex(index=state_county_fip, columns=state_county_fip)
    .fillna(0)
    .sort_index(axis=1)
    .sort_index()
    .astype(int)
)

print(out)

打印:

geoid_d  18001  18005  18039  18050  18111  18182
geoid_o                                          
18001        0      0      0      0      0      0
18005        0      0      0      6      0      0
18039        0      0     10      0      9      0
18050        0      0      0      0      0      0
18111        0      0      0      0      0      0
18182        0      0      0      0      0     15