資料集檢視器文件

DuckDB

Hugging Face's logo
加入 Hugging Face 社群

並獲得增強的文件體驗

開始使用

DuckDB

DuckDB 是一個數據庫,支援非常快速地讀取和查詢 Parquet 檔案。首先,建立與 DuckDB 的連線,然後安裝並載入 httpfs 擴充套件以讀取和寫入遠端檔案。

Python
JavaScript
import duckdb

url = "https://huggingface.co/datasets/tasksource/blog_authorship_corpus/resolve/refs%2Fconvert%2Fparquet/default/train/0000.parquet"

con = duckdb.connect()
con.execute("INSTALL httpfs;")
con.execute("LOAD httpfs;")

現在,您可以編寫並在 Parquet 檔案上執行 SQL 查詢。

Python
JavaScript
con.sql(f"SELECT sign, count(*), AVG(LENGTH(text)) AS avg_blog_length FROM '{url}' GROUP BY sign ORDER BY avg_blog_length DESC LIMIT(5)")
┌───────────┬──────────────┬────────────────────┐
│   sign    │ count_star() │  avg_blog_length   │
│  varchar  │    int64     │       double       │
├───────────┼──────────────┼────────────────────┤
│ Cancer    │        389561206.5212034089743 │
│ Leo       │        354871180.0673767858652 │
│ Aquarius  │        327231152.1136815084192 │
│ Virgo     │        361891117.1982094006466 │
│ Capricorn │        318251102.397360565593 │
└───────────┴──────────────┴────────────────────┘

要查詢多個檔案(例如,如果資料集已分片)

Python
JavaScript
urls = ["https://huggingface.co/datasets/tasksource/blog_authorship_corpus/resolve/refs%2Fconvert%2Fparquet/default/train/0000.parquet", "https://huggingface.co/datasets/tasksource/blog_authorship_corpus/resolve/refs%2Fconvert%2Fparquet/default/train/0001.parquet"]

con.sql(f"SELECT sign, count(*), AVG(LENGTH(text)) AS avg_blog_length FROM read_parquet({urls}) GROUP BY sign ORDER BY avg_blog_length DESC LIMIT(5)")
┌──────────┬──────────────┬────────────────────┐
│   sign   │ count_star() │  avg_blog_length   │
│ varchar  │    int64     │       double       │
├──────────┼──────────────┼────────────────────┤
│ Aquarius │        496871191.417211745527 │
│ Leo      │        538111183.8782219248853 │
│ Cancer   │        650481158.9691612347804 │
│ Gemini   │        519851156.0693084543618 │
│ Virgo    │        603991140.9584430205798 │
└──────────┴──────────────┴────────────────────┘

DuckDB-Wasm 是一個由 WebAssembly 提供支援的包,也可用於在任何瀏覽器中執行 DuckDB。例如,如果您想建立一個 Web 應用程式以從瀏覽器查詢 Parquet 檔案,這可能會很有用!

< > 在 GitHub 上更新

© . This site is unofficial and not affiliated with Hugging Face, Inc.