如何截断Numpy数组输出？实现类似MATLAB的摘要显示

1. 内置的截断输出支持 #

Numpy本身没有直接对应MATLAB的“摘要模式”（仅显示形状和类型），但通过np.set_printoptions()可以配置元素级截断，实现首尾保留、中间省略的效果，比简单截断字符更合理：

import numpy as np

# 配置全局打印规则
np.set_printoptions(
    edgeitems=2,        # 每个维度的首尾各显示2个元素
    threshold=50,       # 数组元素总数超过50时触发截断
    linewidth=80,       # 限制每行输出宽度，避免换行混乱
    formatter={'float': '{:.2f}'.format}  # 可选：统一数值格式
)

# 测试打印大数组字典
arr_dict = {n: np.random.randn(50, 50, 50) for n in 'abcd'}
print(arr_dict)

配置后，大数组会以类似[[[ 0.12, 0.34, ..., -0.56, 0.78], ..., [ 1.23, -4.56, ..., 7.89, 0.10]]]的形式输出，自动省略中间冗余元素。

2. 实现MATLAB风格的简洁摘要 #

如果只需要显示数组的形状和类型（类似MATLAB的"a" ⟼ {50×50×50 double}），可以自定义打印函数，直接输出摘要信息：

def print_array_summary(arr_dict):
    for key, arr in arr_dict.items():
        print(f'"{key}" ⟼ {arr.shape} {arr.dtype}')

print_array_summary(arr_dict)

执行后会输出：

"a" ⟼ (50, 50, 50) float64
"b" ⟼ (50, 50, 50) float64
"c" ⟼ (50, 50, 50) float64
"d" ⟼ (50, 50, 50) float64

3. 字符数限制的替代方案 #

Numpy没有内置的“最大输出字符数”参数，但可以结合np.array2string()和字符串处理，实现更精准的字符截断：

def truncated_str_by_length(arr, max_chars=400):
    arr_str = np.array2string(arr, edgeitems=2, threshold=100)
    if len(arr_str) <= max_chars:
        return arr_str
    # 保留开头和结尾，中间用省略号连接
    return f"{arr_str[:max_chars//2 - 3]} ... {arr_str[-max_chars//2:]}"

for key, arr in arr_dict.items():
    print(f'{key}: {truncated_str_by_length(arr, max_chars=400)}')

这种方式可以确保输出字符数不超过设定值，同时保留数组的结构特征。

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