如何用Colorama为列表元素分配对应颜色并以矩阵形式打印？

首先要确保Colorama正确初始化，这是容易忽略的核心步骤。其次，为目标字符建立颜色映射字典，打印每个字符时先输出对应颜色控制码，最后重置颜色避免后续字符继承颜色。

关键修正点： #

• 必须调用 colorama.init() 初始化Colorama（Windows系统必填，Linux/macOS建议添加）
• 建立字符与Colorama颜色常量的映射（比如{'R': Fore.RED, 'G': Fore.GREEN}）
• 打印字符时先输出颜色码，再输出字符，可通过autoreset=True自动重置颜色

修改后的完整代码示例： #

import random
from colorama import Fore, Style, init

# 初始化Colorama，autoreset=True自动重置后续字符颜色
init(autoreset=True)

# 按需求定义字符-颜色映射
COLOR_MAP = {
    'R': Fore.RED,
    'G': Fore.GREEN,
    'B': Fore.BLUE,
    'Y': Fore.YELLOW
}

def generate_random_string(rows, cols, charset):
    return ''.join(random.choice(charset) for _ in range(rows * cols))

def print_matrix(random_str, rows, cols):
    char_list = list(random_str)
    for i in range(rows):
        # 截取当前行的字符
        row_chars = char_list[i*cols : (i+1)*cols]
        # 为每个字符匹配颜色，无映射则用默认色
        colored_row = [COLOR_MAP.get(char, '') + char for char in row_chars]
        print(' '.join(colored_row))

# 示例调用
if __name__ == "__main__":
    rows = random.randint(3, 5)
    cols = random.randint(3, 5)
    charset = ['R', 'G', 'B', 'Y']
    rand_str = generate_random_string(rows, cols, charset)
    print_matrix(rand_str, rows, cols)

常见问题排查： #

• 未安装Colorama时执行：pip install colorama
• Windows旧版CMD不支持ANSI颜色，可改用init(convert=True)强制转换
• 若手动控制颜色重置，需在每个字符后追加Style.RESET_ALL

内容的提问来源于stack exchange，提问作者Vincent- Vega