import pandas as pd
import datetime

# 读取 CSV 文件
df = pd.read_csv('./All-2024-09-03_15-01-27.csv')

# 指定要查询的 ID 列表
query_ids = [1000002, 1000001, 1000000, 1000003]
print("{0}---{1}".format("查询Ids值",query_ids))

# 创建查询 DataFrame
query_df = pd.DataFrame({'ID': query_ids})

# 根据 ID 查询数据并保留重复项
results = query_df.merge(df, on='ID', how='left')
# 当前时间戳
now = datetime.datetime.now()
timestamp = now.strftime("%Y-%m-%d_%H-%M-%S")
filename = f"Query-IDs.csv"

# 保存到 CSV 文件
results.to_csv(filename, index=False, encoding='utf-8')
print("报表已成功导出为 '" + filename +"'")