Python库巡礼--DuckDB

import duckdb# 创建内存数据库连接con = duckdb.connect(database=':memory:')# 直接执行SQL（自动创建表）con.execute("""    CREATE TABLE sales (        product VARCHAR,        amount DECIMAL(10,2),        date DATE    )""")# 插入数据con.execute("INSERT INTO sales VALUES ('Laptop', 1200.50, '2025-01-15')")con.execute("INSERT INTO sales VALUES ('Mouse', 25.99, '2025-01-16')")# 查询并转为Pandas DataFramedf = con.execute("""    SELECT product, amount     FROM sales     WHERE amount > 100""").fetchdf()print(df)#      product   amount# 0     Laptop  1200.50

import pandas as pdimport duckdb# 创建Pandas DataFramedf = pd.DataFrame({    'name': ['Alice', 'Bob', 'Charlie'],    'age': [25, 30, 35],    'salary': [50000, 60000, 75000]})# 将DataFrame注册为DuckDB虚拟表con = duckdb.connect()con.register('employees', df)# 用SQL分析Pandas数据high_earners = con.sql("""    SELECT name, salary     FROM employees     WHERE salary > 60000""").df()print(high_earners)#      name  salary# 0  Charlie   75000

import duckdb# 查询JSON文件并使用窗口函数result = duckdb.sql("""    WITH user_events AS (        SELECT             user_id,            event_type,            event_time,            ROW_NUMBER() OVER (                PARTITION BY user_id                 ORDER BY event_time DESC            ) as rn        FROM read_json_auto('events.json')    )    SELECT user_id, event_type, event_time    FROM user_events    WHERE rn = 1  -- 每个用户的最新事件""").show()  # 直接在终端显示表格

ROW_NUMBER()