python 筛选目标站点的数据 采用isin函数

isin 函数的使用

返回的结果是根据从isin函数传入的列表(list)，筛选出与列表中包含的数值或字符串相同的数据记录, 用法有点类似sql中的"in"

 原始数据一堆站点，想筛选目标站点

#!usr/bin/env python
# -*- coding:utf-8 -*-
"""
@author: Su
@file: choose_station.py
@time: 2023/08/29
@desc:
"""
import pandas as pd

df = pd.read_excel('/processdata/lianxi/0-20RH.xlsx')
df = pd.DataFrame(df)

list = [50425,53463]
ab = df[df['station'].isin(list)]
print(ab)
# ab.to_excel('processdata/lianxi/lianxi.xlsx', index=False)

结果