numpy.set_string_function

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

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

Параметры

fфункция или None

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

reprbool, необязательно

Если 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])'