### 使用Python实现从Excel文件导入数据到PostgreSQL数据库，并新增一列存储文件名 以下是完整的Python代码示例，该代码实现了从Excel文件读取数据并将其插入到PostgreSQL数据库中的功能。同时，在目标表中新增了一列表示文件名。 #### Python代码示例 ```python import pandas as pd import psycopg2 def insert_excel_to_postgresql(file_path, table_name, database_config): """ 将Excel文件的数据导入到PostgreSQL数据库，并在表中新增一列存储文件名。 :param file_path: Excel文件路径 :param table_name: PostgreSQL的目标表名称 :param database_config: 数据库连接配置字典 """ try: # 读取Excel文件 df = pd.read_excel(file_path) # 新增一列表示文件名 filename_column = 'filename' # 定义新列的名称 df[filename_column] = file_path.split('/')[-1] # 连接PostgreSQL数据库 connection = psycopg2.connect( host=database_config['host'], port=database_config['port'], user=database_config['user'], password=database_config['password'], dbname=database_config['dbname'] ) cursor = connection.cursor() # 构造SQL语句动态插入数据 columns = list(df.columns) placeholders = ','.join(['%s'] * len(columns)) sql_template = f"INSERT INTO {table_name} ({','.join(columns)}) VALUES ({placeholders})" # 批量插入数据 data_tuples = [tuple(row) for row in df.to_numpy()] cursor.executemany(sql_template, data_tuples) # 提交事务 connection.commit() print(f"成功将文件 {file_path} 的数据导入到表 {table_name}") except Exception as e: print(f"发生错误：{e}") finally: if 'cursor' in locals(): cursor.close() if 'connection' in locals() and connection is not None: connection.close() # 配置数据库连接信息 db_config = { 'host': 'localhost', 'port': 5432, 'user': 'your_user', # 替换为实际用户名 'password': 'your_password', # 替换为实际密码 'dbname': 'your_database' # 替换为实际数据库名 } # 调用函数 insert_excel_to_postgresql('example.xlsx', 'target_table', db_config) ``` --- #### 功能解析 1. **读取Excel文件**: 利用 `pandas` 库加载Excel文件的内容[^1]。 2. **新增文件名列**: 在DataFrame对象中增加一列，其值设置为当前Excel文件的名字[^1]。 3. **连接PostgreSQL数据库**: 使用 `psycopg2` 建立与PostgreSQL数据库的连接[^1]。 4. **批量插入数据**: 构建动态SQL语句并通过 `executemany()` 方法高效地完成大批量数据的插入操作[^1]。 5. **异常处理与资源释放**: 确保即使程序运行过程中出现问题也能安全关闭数据库连接和游标。 --- #### 注意事项 - 确保安装必要的依赖库：`pip install pandas psycopg2-binary openpyxl` - 如果目标表不存在或者结构不匹配，请提前手动创建好对应的表结构[^1]。 - 文件路径应为绝对路径或相对于脚本所在目录的有效路径[^1]。 ---