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])'