AcidMap/Water/Python_codes/CSV_Integration.py

import pandas as pd

# 读取原始CSV文件
input_file = r'D:\17417\Documents\backend\Water\Data\四县三用地&水田.csv'
output_file = r'D:\17417\Documents\backend\Water\Data\中心点经纬度与预测值&水田.csv'

try:
    # 读取CSV文件
    df = pd.read_csv(input_file)
    
    # 提取需要的三列数据
    extracted_data = df[['中心点经度', '中心点纬度', 'Cd (ug/L)']]
    
    # 重命名列名
    extracted_data.columns = ['lon', 'lan', 'Prediction']
    
    # 保存到新的CSV文件
    extracted_data.to_csv(output_file, index=False, encoding='utf-8-sig')
    
    print(f"数据已成功提取并保存到: {output_file}")
    
except FileNotFoundError:
    print(f"错误: 文件 {input_file} 未找到，请检查文件路径。")
except KeyError as e:
    print(f"错误: 找不到列 {e}，请检查CSV文件中的列名。")
except Exception as e:
    print(f"发生未知错误: {e}")