Numpy 2.0.0中numpy.float32列表打印格式变化问题咨询

我运行模拟得到以下字典结果：

results = {
    'this': 5,
    'that': 6,
    'those': [2.34, 5.67]
}

其中those对应的列表实际是numpy.float32类型的元素。我用下面的代码将其保存为人类可读格式：

s = ''
for key, value in results.items():
    s += f'{key}: {value}\n'

在numpy 1.26.1时，输出是[2.34, 5.67]，但升级到numpy 2.0.0后，输出变成了[np.float32(2.34), np.float32(5.67)]。

另外测试了几个交互场景：

import numpy
x = numpy.float32(2.3)

# 直接打印x
# numpy 1 — 2.3
# numpy 2 — np.float32(2.3)

# f-string格式化单个x
# numpy 1 — '2.299999952316284'
# numpy 2 — '2.299999952316284'

# f-string格式化包含x的列表
# numpy 1 — '[2.3, 2.3]'
# numpy 2 — '[np.float32(2.3), np.float32(2.3)]'

# 包含x的字典
y={x:x*x}
# numpy 1 — {2.3: 5.29}
# numpy 2 — {np.float32(2.3): np.float32(5.29)}

一、有没有通用解决方案生成可读输出？ #

不需要针对不同类型编写不同代码，有两种通用方案可选：

1. 递归转换numpy标量为Python原生类型 #

写一个递归函数，遍历所有嵌套结构（列表、字典等），将numpy标量转换为对应的Python原生类型：

import numpy as np

def convert_numpy_to_python(obj):
    if isinstance(obj, np.generic):
        return obj.item()
    elif isinstance(obj, list):
        return [convert_numpy_to_python(item) for item in obj]
    elif isinstance(obj, dict):
        return {convert_numpy_to_python(k): convert_numpy_to_python(v) for k, v in obj.items()}
    else:
        return obj

# 使用示例
formatted_results = convert_numpy_to_python(results)
s = ''
for key, value in formatted_results.items():
    s += f'{key}: {value}\n'

处理后numpy.float32会转为Python的float，输出会回到[2.34, 5.67]的可读格式。

2. 自定义格式化函数处理容器和标量 #

针对容器类型单独处理，或借助numpy工具函数统一格式化：

import numpy as np

def format_value(value):
    if isinstance(value, list):
        try:
            arr = np.array(value)
            return np.array2string(arr, separator=', ')
        except:
            return str(value)
    elif isinstance(value, np.generic):
        return str(value.item())
    else:
        return str(value)

# 使用示例
s = ''
for key, value in results.items():
    s += f'{key}: {format_value(value)}\n'

二、这个变化是bug还是特性？ #

这是numpy 2.0的有意特性变更，不是bug，大概率不会回退。

numpy 2.0对标量的字符串表示做了调整：直接打印numpy标量时，会显式展示类型信息（比如np.float32(2.3)），而旧版本会隐式伪装成Python float的样子。同时，列表、字典等容器的默认字符串表示会调用元素的__repr__方法，numpy 2.0修改了标量的__repr__输出逻辑，导致容器的显示也随之变化。

单个标量用f-string格式化时调用的是__str__方法，所以两个版本输出一致；但容器的__str__会迭代调用元素的__repr__，这就是列表、字典显示差异明显的原因。

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