queryIds.py 657 B

1234567891011121314151617181920212223
  1. import pandas as pd
  2. import datetime
  3. # 读取 CSV 文件
  4. df = pd.read_csv('./All-2024-09-03_15-01-27.csv')
  5. # 指定要查询的 ID 列表
  6. query_ids = [1000002, 1000001, 1000000, 1000003]
  7. print("{0}---{1}".format("查询Ids值",query_ids))
  8. # 创建查询 DataFrame
  9. query_df = pd.DataFrame({'ID': query_ids})
  10. # 根据 ID 查询数据并保留重复项
  11. results = query_df.merge(df, on='ID', how='left')
  12. # 当前时间戳
  13. now = datetime.datetime.now()
  14. timestamp = now.strftime("%Y-%m-%d_%H-%M-%S")
  15. filename = f"Query-IDs.csv"
  16. # 保存到 CSV 文件
  17. results.to_csv(filename, index=False, encoding='utf-8')
  18. print("报表已成功导出为 '" + filename +"'")