在实际的工作中,经常遇到“从详细信息列表中,根据关键信息快速查询和定位”的问题。例如如下,根据国家(中国、美国和英国),快速查询相应的详细信息(城市、人口)。
解决方案:利用pandas.index.intersection
具体实现,如下代码
import pandas as pd
# 读入数据
detail=pd.read_csv('详细信息.csv') #国家的详细信息列表
daiChaXun=pd.read_csv('待查询国家.csv') #待查询的国家列表
print(detail, '\n\n', daiChaXun)
# 将详细信息列表、待查询的国家列表的索引,重新设置为‘国家’列
detail.set_index('国家', inplace=True)
daiChaXun.set_index('国家', inplace=True)
print(detail, '\n\n', daiChaXun)
# 利用pandas.index.intersection,求2个列表关于索引列‘国家’的交集
idx=detail.index.intersection(daiChaXun.index)
# 得到待查询国家的详细信息,并把索引恢复成默认值
detail_daiChaXun=detail.loc[idx].reset_index()
print(detail_daiChaXun)
# 恢复详细信息列表、待查询的国家列表的索引
detail.reset_index()
daiChaXun.reset_index()