根据list找到df对应数据的两种常用方法

徐曼巴

已于 2024-03-19 11:45:31 修改

阅读量182

点赞数 2

文章标签： python pandas

于 2024-03-19 11:38:47 首次发布

本文链接：https://blog.csdn.net/qq_53899001/article/details/136837322

版权

文章介绍了如何使用Python中的Pandas库，通过条件查询(cbg_features_df[census_block_group]==a)获取census_block_group对应的人口数据，以及如何使用merge方法合并数据，最终提取出特定列(census_block_group和Totalpopulation)以CSV格式输出。

摘要由CSDN通过智能技术生成

1.condition

import json
import sys
import pandas as pd
pd.set_option('display.max_columns', None)

cbg_features_df = pd.read_csv('cbg_features_2019.csv')

with open('cbg_lonlat_dict.json', 'r', encoding='utf-8') as f:
    cbg_lonlat_dict = json.load(f)

cbg_lonlat_dict_keys_list = list(cbg_lonlat_dict.keys())
int_list = [int(x) for x in cbg_lonlat_dict_keys_list]

cbg_total_population_list = []

for a in int_list:
    condition = cbg_features_df['census_block_group'] == a
    total_population = cbg_features_df[condition]['Total population'].iloc[0]
    cbg_total_population_list.append(total_population)

cbg_activate_df = pd.DataFrame({'cbg_id': cbg_lonlat_dict_keys_list, 'cbg_total_people': cbg_total_population_list})

print(cbg_activate_df)

核心代码其实就是这两行：

condition = cbg_features_df['census_block_group'] == a
total_population = cbg_features_df[condition]['Total population'].iloc[0]

2.merge

import json
import sys
import pandas as pd
pd.set_option('display.max_columns', None)

cbg_features_df = pd.read_csv('cbg_features_2019.csv')

with open('cbg_lonlat_dict.json', 'r', encoding='utf-8') as f:
    cbg_lonlat_dict = json.load(f)

cbg_lonlat_dict_keys_list = list(cbg_lonlat_dict.keys())
int_list = [int(x) for x in cbg_lonlat_dict_keys_list]

cbg_activate_df = pd.merge(pd.DataFrame({'census_block_group': int_list}), cbg_features_df, on='census_block_group', how='left')

cbg_activate_df = cbg_activate_df[['census_block_group', 'Total population']]

print(cbg_activate_df)

可以看到merge方法在做此类问题的时候比较优雅，核心代码列出如下：

cbg_activate_df = pd.merge(pd.DataFrame({'census_block_group': int_list}), cbg_features_df, on='census_block_group', how='left')
cbg_activate_df = cbg_activate_df[['census_block_group', 'Total population']]