使用 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免费学习笔记(深入)”;