Python 读取 Excel 表格并导出为 DBF 文件

以下是将上述代码封装为函数的版本。函数接收 input_excel_path、sheet_name 和 output_dbf_path 作为参数,按照需求读取 Excel 表格并导出为 DBF 文件。

封装函数代码

python 复制代码
import pandas as pd
import dbf

def excel_to_dbf(input_excel_path, sheet_name, output_dbf_path):
    """
    将 Excel 的指定工作表导出为 DBF 文件,保留数据类型和列名格式。

    参数:
        input_excel_path (str): 输入的 Excel 文件路径。
        sheet_name (str): 需要读取的工作表名称。
        output_dbf_path (str): 导出的 DBF 文件路径。
    """
    # 读取 Excel 的指定表
    df = pd.read_excel(input_excel_path, sheet_name=sheet_name)

    # 定义字段字符串列表,根据数据类型进行动态判断
    def get_field_definition(column):
        if df[column].dtype == 'object':  # 字符类型
            return f"{column} C(255)"
        elif pd.api.types.is_integer_dtype(df[column]):  # 整型
            return f"{column} N(10,0)"  # 整型无需小数位,最大宽度10
        elif pd.api.types.is_float_dtype(df[column]):  # 浮点型
            # 判断小数位数
            max_decimal_places = df[column].apply(
                lambda x: len(str(x).split(".")[1]) if pd.notnull(x) and "." in str(x) else 0
            ).max()
            return f"{column} N(18,{min(max_decimal_places, 5)})"  # 浮点型设置小数位,限制最大5位
        elif pd.api.types.is_datetime64_any_dtype(df[column]):  # 日期类型
            return f"{column} D"
        else:
            raise ValueError(f"Unsupported column type for {column}")

    # 生成字段定义,确保列名格式不变
    field_definitions = ";".join([get_field_definition(col) for col in df.columns])

    # 使用指定编码创建 DBF 表
    table = dbf.Table(output_dbf_path, field_definitions, codepage="utf8")
    table.open(dbf.READ_WRITE)

    # 写入数据
    for _, row in df.iterrows():
        row_data = tuple(row.fillna("").to_list())  # 用空值填充 NaN
        table.append(row_data)

    table.close()
    print(f"The table {sheet_name} exported to {output_dbf_path}.")

# 使用示例
if __name__ == "__main__":
    input_excel_path = r"E:\_OrderingProject\F_IslandsBoundaryChange\c_GeeData\Tif_Thumbnail_check\_check.xlsx"
    sheet_name = "fix_v1"
    output_dbf_path = r"E:\_OrderingProject\F_IslandsBoundaryChange\c_GeeData\Tif_Thumbnail_check\fix_v1.dbf"
    
    excel_to_dbf(input_excel_path, sheet_name, output_dbf_path)

函数说明

  1. 函数参数:

    • input_excel_path:输入的 Excel 文件路径。
    • sheet_name:Excel 文件中需要导出的工作表名称。
    • output_dbf_path:目标 DBF 文件的输出路径。
  2. 动态字段定义:

    • 根据列数据类型(字符、整型、浮点型、日期等)生成适当的 DBF 字段定义。
    • 确保小数位数动态调整(浮点型最多保留 5 位小数)。
  3. 自动填充缺失值:

    • 用空字符串填充 NaN 值,确保写入 DBF 时不会出错。
  4. 编码:

    • 使用 utf8 编码创建 DBF 表,保证支持中文字符。
  5. 异常处理:

    • 如果遇到不支持的数据类型,会引发 ValueError,提示用户检查输入数据。

示例输出

假设输入的 fix_v1 工作表数据如下:

ID Name Score Date
1 Alice 95.0 2024-12-09
2 Bob 89 2024-12-08
3 Carol 82.5 2024-12-07

导出的 fix_v1.dbf 文件会保留字段类型和格式一致。

如果这对您有所帮助,希望点赞支持一下作者! 😊

详细全文-点击查看

相关推荐
维克兜率天12 分钟前
【维克】趋势型CTA:期货市场的“追涨杀跌“
人工智能·python
Zenova EdgeOS1 小时前
Go 工业边缘错误处理实战:用 errors 构建可追溯错误链
开发语言·数据库·golang
取名真是1 小时前
Python 算法知识点从基础到进阶
开发语言·python·算法
梦想不只是梦与想1 小时前
PyTorch、CPU、GPU、CUDA
人工智能·pytorch·python·深度学习·机器学习·cuda
数字融合1 小时前
透明化视频孪生危化安全管控解决方案
人工智能·python·安全
FRing031 小时前
vue2干货总结3(computed计算属性、watch监听器)
开发语言·前端·vue.js·vue2
Fatbobman(东坡肘子)2 小时前
快速消失的硬盘容量 -- 肘子的 Swift 周报 #160
开发语言·人工智能·redis·ios·swiftui·iphone·swift
揽星河..3 小时前
【C语言】底层程序设计
c语言·开发语言
JavaPub-rodert3 小时前
从零搭建通用管理后台:ShiyuAdmin 的 Go、React 与 RBAC 实践
开发语言·react.js·golang
别问!问就是全会3 小时前
Jupyter Notebook 安装包下载|单元格分块运行,图表直接内嵌
ide·python·其他·jupyter