使用 Pandas read_sql 传递参数

概述

Pandas read_sql 函数允许您执行 SQL 查询并从数据库检索数据。它的功能之一是能够将参数传递给查询。

参数语法

参数可以作为列表、元组或字典传递。确切的语法取决于所使用的数据库驱动程序。以下是一些常见的示例：

• ？或 %s：单个值的占位符
• :1 或 :name：单个值的命名占位符
• %(name)s：使用 string.format 语法的命名占位符

使用带有命名参数的字典

传递参数的一个选项是使用字典。大多数驱动程序都支持这一点，包括带有 psycopg2 驱动程序的 PostgreSQL。字典中的键值对对应于查询中的命名占位符。

示例

为了演示命名参数方法，让我们考虑以下 SQL 查询：

select "Timestamp", "Value" from "MyTable"
where "Timestamp" BETWEEN :dstart AND :dfinish

以下是使用字典将参数传递给此查询的方法：

params = {"dstart": datetime(2014, 6, 24, 16, 0), "dfinish": datetime(2014, 6, 24, 17, 0)}

df = psql.read_sql(
    "select \"Timestamp\",\"Value\" from \"MyTable\" where \"Timestamp\" BETWEEN %(dstart)s AND %(dfinish)s",
    db,
    params=params,
    index_col=["Timestamp"],
)

在此示例中，params 字典为命名占位符 :dstart 和 :dfinish 提供值。 %(name)s 语法确保将值正确插入到查询中。