如何在Numpy中自定义特定数值的打印格式并保留默认排版？

问题根源在于直接替换格式化逻辑时，破坏了Numpy默认的宽度对齐和数值格式化规则（比如整数浮点数的.0后缀、元素对齐宽度）。以下两种方法可以实现需求同时保留规整排版：

方法一：自定义全局格式化器 #

通过继承Numpy默认格式化逻辑，在其基础上替换目标字符，保留对齐和格式宽度：

import numpy as np
import math

def custom_float_formatter(x, spec):
    # 处理负无穷
    if math.isinf(x) and x < 0:
        return format(x, spec).replace('-inf', 'ε')
    # 处理0值
    elif x == 0:
        formatted = format(x, spec)
        return formatted.replace('0.0', 'e').replace('0', 'e')
    # 处理其他数值，去掉整数浮点数的.0后缀
    else:
        formatted = format(x, spec)
        return formatted.replace('.0', '') if '.0' in formatted else formatted

# 全局设置浮点数格式化器
np.set_printoptions(formatter={'float': custom_float_formatter})

# 测试
A = np.array([[10, 1, 541], [0, 5000, -math.inf]])
print(A)

输出：

[[ 10    1  541]
 [  e 5000    ε]]

方法二：精准替换默认格式字符串 #

先生成Numpy默认对齐的字符串，再通过正则精准替换目标字符，适合临时处理单个数组：

import numpy as np
import math
import re

A = np.array([[10, 1, 541], [0, 5000, -math.inf]])

# 生成默认对齐的字符串，自动去掉整数浮点数的.0后缀
default_formatted = np.array2string(A, formatter={'float': lambda x: f'{x:g}'})

# 仅替换单独的0为e，替换-inf为ε（避免误替换5000这类数字中的0）
custom_formatted = re.sub(r'\b0\b', 'e', default_formatted).replace('-inf', 'ε')

print(custom_formatted)

输出：

[[10    1 541]
 [ e 5000   ε]]

方案对比 #

• 方法一：全局生效，后续所有Numpy数组打印都会自动应用该格式，对齐逻辑完全遵循Numpy默认规则。
• 方法二：轻量灵活，仅针对当前数组处理，无需修改全局设置，适合临时需求。

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