php中文网

Python DuckDB中如何用read_csv函数指定CSV文件的字段类型?

php中文网

使用 read_csv 导入 csv 指定字段类型

在 python 中使用 duckdb sdk 读取 csv 文件时,可以通过 read_csv() 函数指定列的字段类型以避免类型转换错误。该函数允许在导入时设置特定列的数据类型。

示例代码:

import duckdb

# 指定列类型映射
types = {'column_name': 'data_type'}

# 读取 CSV 文件并指定列类型
query = f"""
SELECT *
FROM read_csv('{csv_file_path}',
    types = {types});
"""
df = duckdb.query(query).df()

通过在 types 字典中指定列名和相应的数据类型,可以确保在导入期间不出现类型转换错误。

立即学习“Python免费学习笔记(深入)”;

以上就是Python DuckDB中如何用read_csv函数指定CSV文件的字段类型?的详细内容,更多请关注php中文网其它相关文章!