找回密码
 立即注册
软件设计/软件工程 2022-05-05 230 0star收藏 版权: . 保留作者信息 . 禁止商业使用 . 禁止修改作品
问题
我正在尝试将 pandas 数据框写入 Postgres 数据库。

代码显示如下:
  1. dbConnection = psycopg2.connect(user = "user1", password = "user1", host = "localhost", port = "5432", database = "postgres")
  2. dbConnection.set_isolation_level(0)
  3. dbCursor = dbConnection.cursor()
  4. dbCursor.execute("DROP DATABASE IF EXISTS FiguresUSA")
  5. dbCursor.execute("CREATE DATABASE FiguresUSA")
  6. dbCursor.execute("DROP TABLE IF EXISTS FiguresUSAByState")
  7. dbCursor.execute("CREATE TABLE FiguresUSAByState(Index integer PRIMARY KEY, Province_State VARCHAR(50), NumberByState integer)");

  8. for i in data_pandas.index:
  9.     query = """
  10.     INSERT into FiguresUSAByState(column1, column2, column3) values('%s',%s,%i);
  11.     """ % (data_pandas['Index'], data_pandas['Province_State'], data_pandas['NumberByState'])

  12. dbCursor.execute(query)
复制代码

当我运行它时,我得到一个错误,它只是说:“索引”。我知道这是我的 for 循环中某处的问题,这个 % 符号是否正确?我是 Postgres 的新手,不知道它怎么可能是正确的语法。任何人都可以帮忙吗?我知道我可以使用 sql,但我正在尝试一种不同的技术。

数据打印如下:

一个轻微的例外是有一个“索引”。在 IDE 版本中。这可能是问题吗?

回答
如果使用 pd.DataFrame.to_sql,您可以提供 index_label 参数以将其用作列。

data_pandas.to_sql('FiguresUSAByState', con=dbConnection, index_label='Index')

如果您希望坚持使用现有的自定义 SQL 和 for 循环,则需要先重置索引。

  1. for row in data_pandas.reset_index().to_dict('rows'):
  2.     query = """
  3.     INSERT into FiguresUSAByState(index, Province_State, NumberByState) values(%i, '%s', %i);
  4.     """ % (row['index'], row['Province_State'], row['NumberByState'])
复制代码

请注意,新列的默认名称是未大写的 index ,而不是 Index 。





上一篇:添加格式为 dd:HH:mm:ss 的日期
下一篇:在 JavaScript 中解析 JSON 对象数组