Функция option_context

Функция option_context применяется для временной установки опций Pandas в пределах контекстного менеджера. После выхода из контекста все опции автоматически возвращаются к исходным значениям. Функция принимает пары параметров в виде строки с именем опции и нового значения.

Синтаксис

with pd.option_context('опция', значение, ...):
	# код с временными настройками

Пример

Давайте изменим количество отображаемых строк и точность чисел:

import pandas as pd

df = pd.DataFrame({
	'a': [1.123456, 2.123456, 3.123456],
	'b': [4.123456, 5.123456, 6.123456]
})

with pd.option_context('display.max_rows', 2, 'display.precision', 2):
	print(df)

Результат выполнения кода:

 a	b
		0  1.12  4.12
		1  2.12  5.12
		2  3.12  6.12

Пример

Давайте изменим количество отображаемых столбцов и ширину вывода:

import pandas as pd

df = pd.DataFrame({
	'a': [1, 2, 3],
	'b': [4, 5, 6],
	'c': [7, 8, 9],
	'd': [10, 11, 12],
	'e': [13, 14, 15]
})

with pd.option_context('display.max_columns', 3, 'display.width', 50):
	print(df)

Результат выполнения кода:

   a  b  c ...  d   e
0  1  4  7 ... 10  13
1  2  5  8 ... 11  14
2  3  6  9 ... 12  15

[3 rows x 5 columns]

Пример

Давайте изменим формат отображения чисел с плавающей точкой:

import pandas as pd

df = pd.DataFrame({
	'a': [0.123456, 0.234567, 0.345678],
	'b': [0.456789, 0.567890, 0.678901]
})

with pd.option_context('display.float_format', '{:.2f}'.format):
	print(df)

Результат выполнения кода:

 a	b
		0  0.12  0.46
		1  0.23  0.57
		2  0.35  0.68

Смотрите также

функция set_option,
которая устанавливает глобальные опции
функция get_option,
которая возвращает текущее значение опции
функция reset_option,
которая сбрасывает опции к значениям по умолчанию
функция describe_option,
которая выводит описание опции