Pandas | Sqlalchemy 与MySQL的坑

问题需求：

Pandas处理数据之后写回到MySQL数据库时报了一个错误。

报错示例：

sqlalchemy.exc.ProgrammingError: (MySQLdb._exceptions.ProgrammingError) inf can not be used with MySQL
or
ValueError: inf cannot be used with MySQL

产生原因：

pandas 数据处理过程中用到了除法(除数为0)，导致出现inf，而Mysql数据库不支持写入该值。

解决办法：

将处理过之后的DataFrame中的inf值替换掉(np.inf为正无穷，-np.inf为负无穷)。

解决示例：

# -*- encoding: utf-8 -*-
"""
@Author   : Ccm
@Time     : 2022-01
"""

import os
import pandas as pd
from sqlalchemy import create_engine

engine = config.Config.SQLALCHEMY_DATABASE_URI

file_dir = "static/related_tables/entrydata/elects"
tables_name = "elects"
df = pd.read_excel(os.path.join(file_dir, single_file))
df_all = df.replace([np.inf, -np.inf], np.nan)
df_all.to_sql(tables_name, engine, if_exists=add_type)