Pandas 合并 DataFrame：根据站点号添加 Flag 列

使用 Pandas 合并 DataFrame 添加 Flag 列

本文将介绍如何使用 Pandas 库的 merge 函数，将两个 DataFrame 合并并添加 Flag 列。假设您有一个 DataFrame df1，其中记录了站点号 (station) 以及其对应的 Flag。现在您有一个新的、较大的 CSV 文件 csv_file，要根据该文件中的站点列新建 Flag 列，其 Flag 值应参照 df1 中站点号和 Flag 的对应关系。

步骤：

1. 读取 CSV 文件:

   import pandas as pd
   
   df2 = pd.read_csv('csv_file.csv')
   

2. 合并 DataFrame:

   df3 = pd.merge(df2, df1, on='station', how='left')
   

   使用 merge 函数将 df1 和 df2 合并，根据 'station' 列进行匹配。how='left' 表示只保留 df2 中的站点号，如果 df2 中的站点号在 df1 中找不到对应的 Flag，则使用 NaN 填充。

3. 保存结果:

   df3.to_csv('new_csv_file.csv', index=False)
   

   将合并后的 DataFrame df3 保存为一个新的 CSV 文件 new_csv_file.csv。

最终，您将得到一个新的 CSV 文件 new_csv_file.csv，其中包含站点号和根据 df1 中的对应关系新建的 Flag 列。