duckdb如何使用extension以及直接连接数据库

最新推荐文章于 2025-06-25 15:05:20 发布

原创最新推荐文章于 2025-06-25 15:05:20 发布 · 1.1k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#数据库 #python #开发语言

本文介绍了如何在Python中通过DuckDB连接Postgres数据库，并展示了如何安装、加载扩展以及在SQL中使用自定义函数的例子。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

duckdb有大量拓展,本文用postgres这拓展为例子,展示在py代码中如何使用extension,这postgres的作用就是直连pg数据库. 直接上代码:

url = "postgres://postgres:pwd@0.0.0.0:5432/dbname"
con = duckdb.connect()
con.install_extension("postgres")
con.load_extension("postgres")
def add_name(k: str) -> str:
    print (111,k)
    return k + "just test to use func"
con.create_function("add_name", add_name)

a = con.sql(f"SELECT add_name(field1),field2 FROM postgres_scan('{url}', 'public', 'tablename');")
print(a.df)