如何在Jupyter中全局设置矩阵输出的科学计数法乘数格式

在Jupyter环境中使用NumPy生成向量与矩阵时，需要实现以下输出格式要求：

• 矩阵元素输出时小数点前仅保留1位数字（如x.2、x.8，而非多位数整数加小数形式）
• 自动在矩阵前添加科学计数法形式的全局乘数（如1e6，对应数学中的$1×10^6$）
• 希望在代码开头完成全局设置，无需每次调用函数即可自动应用于所有矩阵输出

此前尝试用np.set_printoptions配置格式，但未达到预期效果，当前输出示例：

[[  888888.8888889        0.         1333333.3333333        0.               0.               0.       ]
 [       0.        33333333.3333333        0.               0.               0.               0.       ]
 [ 1333333.3333333        0.         2666666.6666667        0.               0.               0.       ]
 [       0.               0.               0.               0.               0.               0.       ]
 [       0.               0.               0.               0.               0.               0.       ]
 [       0.               0.               0.               0.               0.               0.       ]]

理想输出效果：
矩阵整体带有一个全局缩放乘数（如1e6），所有元素均以小数点前1位+小数部分的形式显示，例如：

1e6 * [[0.9, 0.0, 1.3, 0.0, 0.0, 0.0],
       [0.0, 33.3, 0.0, 0.0, 0.0, 0.0],
       [1.3, 0.0, 2.7, 0.0, 0.0, 0.0],
       [0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
       [0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
       [0.0, 0.0, 0.0, 0.0, 0.0, 0.0]]

要实现这种全局格式化效果，需要自定义NumPy的打印格式化逻辑，结合np.set_printoptions的formatter参数，同时计算全局缩放因子。以下是可直接运行的全局配置代码：

import numpy as np

class MatrixScalerFormatter:
    def __init__(self):
        self.scale = None
    
    def __call__(self, arr):
        # 处理二维及以上数组，向量可自行调整逻辑
        if arr.ndim < 2:
            return np.array2string(arr, formatter={'float': lambda x: f"{x:.1f}"})
        
        # 计算非零元素的最大数量级，确定缩放因子
        non_zero = arr[arr != 0]
        if len(non_zero) == 0:
            self.scale = 1
            return np.array2string(arr, formatter={'float': lambda x: f"{x:.1f}"})
        
        max_magnitude = np.max(np.abs(non_zero))
        self.scale = 10 ** np.floor(np.log10(max_magnitude))
        scaled_arr = arr / self.scale
        
        # 格式化缩放后的数组，元素保留小数点后1位
        arr_str = np.array2string(scaled_arr, formatter={'float': lambda x: f"{x:.1f}"})
        # 添加全局乘数前缀
        return f"{self.scale:.0e} * {arr_str}"

# 全局设置NumPy的数组打印格式
np.set_printoptions(formatter={'all': MatrixScalerFormatter()})

代码说明 #

1. 自定义格式化类：MatrixScalerFormatter会自动计算数组的全局缩放因子，将所有元素缩放到小数点前仅1位的范围
2. 缩放逻辑：通过非零元素的最大数量级确定1eX形式的缩放因子，把数组元素除以该因子后格式化
3. 全局生效：通过np.set_printoptions将自定义格式化器设为全局默认，后续所有NumPy数组输出都会自动应用该格式

测试示例 #

运行以下代码验证效果：

# 生成测试矩阵
test_matrix = np.array([
    [888888.8888889, 0., 1333333.3333333, 0., 0., 0.],
    [0., 33333333.3333333, 0., 0., 0., 0.],
    [1333333.3333333, 0., 2666666.6666667, 0., 0., 0.],
    [0., 0., 0., 0., 0., 0.],
    [0., 0., 0., 0., 0., 0.],
    [0., 0., 0., 0., 0., 0.]
])

print(test_matrix)

输出会符合预期：

1e+06 * [[0.9 0.0 1.3 0.0 0.0 0.0]
 [0.0 33.3 0.0 0.0 0.0 0.0]
 [1.3 0.0 2.7 0.0 0.0 0.0]
 [0.0 0.0 0.0 0.0 0.0 0.0]
 [0.0 0.0 0.0 0.0 0.0 0.0]
 [0.0 0.0 0.0 0.0 0.0 0.0]]

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