如何通过np.printoptions控制整个NumPy数组的字符串表示？

问题根源 #

你之前的代码触发无限递归，是因为arr_trunc里调用np.array2string(x)时，会再次触发printoptions中配置的formatter['array']回调，形成循环调用。

可行解决方案 #

1. 无递归的自定义Formatter

直接构造数组的截断字符串，避免调用np.array2string，从根源上防止递归：

import numpy as np

def truncate_array(arr):
    # 自定义截断逻辑：打印形状+第一行前3个元素（可按需调整）
    shape = arr.shape
    if len(shape) >= 2:
        first_row = arr[0][:3]
        row_content = ", ".join(f"{val:.4f}" for val in first_row) + ", ..."
        return f"ndarray(shape={shape}, first_row=[{row_content}])"
    # 一维/标量数组沿用默认格式化
    return str(arr)

# 配置printoptions，针对所有类型判断是否为数组
with np.printoptions(formatter={'all': lambda x: truncate_array(x) if isinstance(x, np.ndarray) else str(x)}):
    dc = {n: np.random.randn(50, 50) for n in 'abcd'}
    print(dc)

2. 临时猴子补丁（推荐）

通过上下文管理器临时替换np.ndarray.__str__，既实现直接print嵌套对象的需求，又不会永久修改NumPy内部：

import numpy as np
from contextlib import contextmanager

@contextmanager
def truncated_array_print():
    # 保存原始__str__方法
    original_str = np.ndarray.__str__
    
    def custom_str(arr):
        shape = arr.shape
        if len(shape) >= 2:
            # 只打印形状和第一行开头
            first_segment = arr[0][:3].tolist()
            return f"np.ndarray(shape={shape}, first_row={first_segment}...)"
        # 非二维数组沿用原逻辑
        return original_str(arr)
    
    # 临时替换
    np.ndarray.__str__ = custom_str
    try:
        yield
    finally:
        # 恢复原方法
        np.ndarray.__str__ = original_str

# 使用示例
dc = {n: np.random.randn(50, 50) for n in 'abcd'}
with truncated_array_print():
    print(dc)

3. 递归转换嵌套对象

如果不想修改NumPy内部，可通过递归函数自动处理嵌套结构中的数组，再打印转换后的对象：

import numpy as np

def process_nested(obj):
    if isinstance(obj, np.ndarray):
        return f"array(shape={obj.shape}, first_row={obj[0][:3].tolist()}...)"
    elif isinstance(obj, dict):
        return {k: process_nested(v) for k, v in obj.items()}
    elif isinstance(obj, (list, tuple)):
        return type(obj)(process_nested(item) for item in obj)
    else:
        return obj

dc = {n: np.random.randn(50, 50) for n in 'abcd'}
print(process_nested(dc))

方案对比 #

• 自定义Formatter：无需修改类方法，但需确保格式化逻辑不触发递归，适合简单场景。
• 临时猴子补丁：对用户完全透明，直接print原对象即可，上下文管理器保证安全性。
• 递归转换函数：最安全，不触碰NumPy内部，但需要显式调用转换函数。

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