如何将textwrap与Python文本单词语色高亮功能结合使用

你的核心问题主要有两个：一是着色函数输出时没有换行导致所有内容挤在一行，二是textwrap可能会随意拆分单词。结合你只能用ANSI转义序列的限制，我给你整理了针对性的修复方案：

1. 重构着色函数，解决换行与颜色控制问题 #

你的原函数直接逐个打印单词并使用end=" "，但处理完一行后没有添加换行操作，而且颜色重置逻辑也有疏漏（可能导致后续文本持续显示绿色）。我建议把函数改成返回着色后的字符串，这样能更灵活地控制输出：

def highGreen(text, words):
    highlight_set = set(words)
    colored_parts = []
    for word in text.split():
        if word in highlight_set:
            # 给目标单词添加绿色高亮，结尾立即重置颜色
            colored_parts.append(f"\033[32m{word}\033[0m")
        else:
            colored_parts.append(word)
    return " ".join(colored_parts)

这个版本的优势：

• 避免了直接打印带来的换行失控问题
• 确保每个高亮单词后都会重置颜色，不会影响后续文本
• 返回的字符串可以直接对接textwrap或其他文本处理逻辑

2. 正确结合textwrap，避免单词拆分 #

直接对带ANSI转义序列的字符串使用textwrap会导致长度计算错误（转义序列会被当作普通字符），所以正确的流程是先对原始文本进行换行，再给每行着色：

import textwrap

text = "This bed is super uncomfortable."
# 对原始文本换行，设置break_long_words=False禁止拆分完整单词
wrapped_lines = textwrap.wrap(text, width=20, break_long_words=False)

for line in wrapped_lines:
    # 对每行着色后打印，自动实现换行
    print(highGreen(line, ["bed"]))

这里的break_long_words=False是关键，它会让textwrap保留完整单词，不会把长单词拆成两半。你可以根据控制台宽度调整width参数，适配你的输出环境。

3. 额外优化：处理带标点的单词 #

如果你的文本中单词带有标点（比如例子里的uncomfortable.），原函数的匹配逻辑会因为标点无法正确识别单词。可以添加小处理去除单词两端的标点：

import string

def highGreen(text, words):
    highlight_set = set(words)
    colored_parts = []
    for word in text.split():
        # 去除单词两端的标点符号，不改变原单词的显示
        cleaned_word = word.strip(string.punctuation)
        if cleaned_word in highlight_set:
            colored_parts.append(f"\033[32m{word}\033[0m")
        else:
            colored_parts.append(word)
    return " ".join(colored_parts)

这样即使单词带逗号、句号等标点，也能正确匹配并高亮。

内容的提问来源于stack exchange，提问作者AlanaG