numpy.set_string_function

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

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

Параметры:

f : функция или None Функция, используемая для красивого вывода массивов. Функция должна принимать один массив в качестве аргумента и возвращать строку с представлением массива. Если None, функция сбрасывается до значения по умолчанию для NumPy при выводе массивов. repr : bool, необязательно Если 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?
>>> print(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])'