office办公python代码:文本仿表格对齐显示

# -*- coding: utf-8 -*-# @Author : 小红牛# 微信公众号：wdPythondef _format_table_to_text(table_data, dates=None, title=""):    """    将表格数据格式化为对齐的文本。    如果 dates 不为空，则 table_data 的每一行第一个元素为行标签，后续对应 dates 各列。    如果 dates 为空，则 table_data 第一行视为表头，其余为数据行。    """    # 构建二维字符串列表    rows = []    if dates:        # 有日期列：表头为 "类型" + 日期        header = ["类型"] + dates        rows.append(header)        for row in table_data:            cells = []            for cell in row:                if isinstance(cell, list):                    cells.append(";".join(str(c) for c in cell))                else:                    cells.append(str(cell) if cell is not None else "")            rows.append(cells)    else:        # 无日期列：table_data 已经包含表头行（第一行）        for row in table_data:            cells = []            for cell in row:                if isinstance(cell, list):                    cells.append(";".join(str(c) for c in cell))                else:                    cells.append(str(cell) if cell is not None else "")            rows.append(cells)    # 计算每列显示宽度（中文字符按2计）    def display_width(s):        width = 0        for ch in s:            if ord(ch) > 127:   # 非 ASCII 字符（中文等）                width += 2            else:                width += 1        return width    col_count = len(rows[0])    col_widths = [0] * col_count    for row in rows:        for i, cell in enumerate(row):            w = display_width(cell)            if w > col_widths[i]:                col_widths[i] = w    # 左右各加一个空格作为边距    col_widths = [w + 2 for w in col_widths]    lines = []    if title:        lines.append(title)        lines.append("-" * 120)   # 分隔线    # 格式化每行    for row in rows:        line = ""        for i, cell in enumerate(row):            current_width = display_width(cell)            padding = col_widths[i] - current_width            line += cell + " " * padding        lines.append(line.rstrip())    return "\n".join(lines)# ========== 示例使用 ==========if __name__ == "__main__":    # ---------- 示例1：带日期列 ----------    print("【示例1：带日期列（每月涨跌前5）】\n")    dates = ["2024-01", "2024-02", "2024-03"]    data1 = [        ["板块A", 10, 8, 12],        ["板块B", 5, 7, 9],        ["板块C", 15, 6, 11],    ]    title1 = "各板块每月涨幅值"    text1 = _format_table_to_text(data1, dates=dates, title=title1)    print(text1)    print("\n" + "="*80 + "\n")    # ---------- 示例2：普通表格（含表头） ----------    print("【示例2：普通表格（股票查询结果）】\n")    data2 = [        ["股票代码", "股票名称", "月份", "涨幅(%)", "胜率(%)"],        ["000001", "平安银行", 1, 5.2, 80.0],        ["000002", "万科A", 1, 3.8, 70.5],        ["000003", "深发展", 2, 6.1, 85.2],    ]    title2 = "回测数据查询结果（共3条）"    text2 = _format_table_to_text(data2, dates=None, title=title2)    print(text2)