Python numpy set_string_function用法及代码示例

本文简要介绍 python 语言中 numpy.set_string_function 的用法。

用法:

numpy.set_string_function(f, repr=True)

设置漂亮打印数组时要使用的 Python 函数。

参数：

f： 函数或无

用于漂亮打印数组的函数。该函数应该期望一个数组参数并返回数组表示形式的字符串。如果为“无”，则该函数将重置为默认的 NumPy 函数以打印数组。

repr： 布尔型，可选

如果为 True(默认)，则设置漂亮打印的函数(__repr__)，如果为 False，则设置返回默认字符串表示的函数(__str__)。

例子：

>>> def pprint(arr):
...     return 'HA! - What are you going to do now?'
...
>>> np.set_string_function(pprint)
>>> a = np.arange(10)
>>> a
HA! - What are you going to do now?
>>> _ = a
>>> # [0 1 2 3 4 5 6 7 8 9]

我们可以将函数重置为默认值：

>>> np.set_string_function(None)
>>> a
array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])

repr影响漂亮的打印或正常的字符串表示。注意__repr__仍然受设置影响__str__因为返回字符串中每个数组元素的宽度变得等于结果的长度__str__().

>>> x = np.arange(4)
>>> np.set_string_function(lambda x:'random', repr=False)
>>> x.__str__()
'random'
>>> x.__repr__()
'array([0, 1, 2, 3])'