今天遇到一个问题,需要将DataFrame中数据格式化为sql语句
df = pd.DataFrame([{'a':1, 'b':2, 'c':'3'}])
sql = "UPDATE table_name SET a=1, b=2, c='3' WHERE id=1"
解决方案:
>>> import pandas as pd
>>> df = pd.DataFrame([{'a':1, 'b':2, 'c':'3'}])
# 将df转为dict
>>> dic = df.to_dict('records')
# 进行格式化
# 存在两种情况:数值型和字符型,字符型需要加引号,所以要进行判断
>>> data = ', '.join(
"{}='{}'".format(key, value) \
if type(value) == str else "{}={}".format(key, value) \
for key, value in dic[0].items()
)
>>> sql = "UPDATE table_name SET {} WHERE id=1'".format(data)