如何为控制台输出的numpy数组添加颜色标识：正数标绿负数标红？

嘿，这个需求真的很实用！要快速区分Numpy数组里的正负值，完全可以基于你找到的ANSI转义码来实现，我给你整理了一套清晰的可落地方案：

核心思路 #

利用ANSI控制台颜色转义码，给正数、负数、零分别套上不同的颜色标记，遍历数组逐个处理元素后再按原格式输出，一定要记得添加结束颜色的转义码，不然后续控制台的所有输出都会保持当前颜色，这点很重要！

具体实现代码 #

首先定义常用的颜色常量，然后写一个通用的打印函数：

import numpy as np

# 定义ANSI颜色转义码（可选其他颜色，比如亮黄、亮蓝）
RED = '\033[91m'       # 亮红色，用于标记负数
GREEN = '\033[92m'     # 亮绿色，用于标记正数
END_COLOR = '\033[0m'  # 重置为控制台默认颜色

def colored_array_print(arr, decimal_places=2):
    """带颜色打印Numpy数组，区分正负值"""
    # 把多维数组展平，方便逐个处理元素
    flattened_arr = arr.flatten()
    colored_elements = []
    
    for num in flattened_arr:
        # 格式化数值，保留指定小数位数
        num_str = f"{num:.{decimal_places}f}"
        if num > 0:
            colored_elements.append(f"{GREEN}{num_str}{END_COLOR}")
        elif num < 0:
            colored_elements.append(f"{RED}{num_str}{END_COLOR}")
        else:
            # 零用控制台默认颜色输出
            colored_elements.append(num_str)
    
    # 根据原数组的维度，还原输出格式
    if arr.ndim == 1:
        print(' '.join(colored_elements))
    elif arr.ndim == 2:
        cols = arr.shape[1]
        # 按行分割元素并打印
        for i in range(0, len(colored_elements), cols):
            print(' '.join(colored_elements[i:i+cols]))
    # 如果需要支持3维及以上数组，可以在这里扩展逻辑

测试示例 #

调用函数测试一下效果：

# 创建一个测试用的二维浮点数组
test_array = np.array([[2.34, -5.67, 0.0], [-8.91, 0.12, 4.56]])
# 调用带颜色打印函数，保留2位小数
colored_array_print(test_array)

运行后，控制台里正数会显示亮绿色，负数显示亮红色，零保持默认颜色，一眼就能区分开！

补充说明 #

• 兼容性：现代终端（Windows Terminal、Linux/macOS自带终端）都支持ANSI转义码，如果是旧版Windows CMD，可能需要安装colorama库，导入后调用colorama.init()就能兼容。
• 自定义颜色：你可以替换颜色转义码，比如用\033[93m亮黄色标记零，或者用\033[31m深红色代替亮红色，根据自己的视觉习惯调整。
• 性能：如果是超大型数组，遍历处理会有一点点开销，但一般控制台输出的数组不会太大，完全可以忽略这个问题。

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