numpy.set_string_function

numpy.set_string_function(f, repr=True) [source]

Установите функцию Python для использования при красивой печати массивов.

Параметры

ffunction or None

Функция, используемая для красивого вывода массивов. Функция должна принимать один массив в качестве аргумента и возвращать строку с представлением массива. Если None, функция сбрасывается до значения по умолчанию для печати массивов NumPy.

reprbool, optional

Если True (по умолчанию), устанавливается функция для красивого вывода (__repr__), если False, устанавливается функция, возвращающая стандартное строковое представление (__str__).

См. также

set_printoptions, get_printoptions

Примеры

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