Метод pandas DataFrame head по умолчанию возвращает первые 5 строк. Это контролируется параметром n. В этом трюке мы будем использовать partialmethod из стандартной библиотеки functools, чтобы установить n на другое число. Этот трюк доступен на YouTube-канале Dunder Data (подписывайтесь!).

Стать экспертом

Если вы хотите, чтобы вам доверяли принимать решения с помощью панд, вы должны стать экспертом. Я полностью освоил pandas и разработал курсы и упражнения, которые значительно улучшат ваши знания и эффективность при анализе данных.

  • Анализ основных данных с помощью Python - мой комплексный курс с 800+ страницами, 350+ упражнениями, несколькими проектами и подробными решениями, которые помогут вам стать экспертом в pandas.
  • Получите образец материала, записавшись на бесплатный курс Введение в Pandas.

Сначала давайте прочитаем образец DataFrame, содержащий велосипедные поездки из города Чикаго, и вызовем метод head со значениями по умолчанию. Обратите внимание, что он возвращает 5 строк.

import pandas as pd
bikes = pd.read_csv('data/bikes.csv')
bikes.head()

Стандартная библиотека functools поставляется вместе с partialmethod, которая позволяет вам устанавливать параметры определенного метода. С его помощью вы можете установить любое количество параметров. Ниже мы переназначаем метод DataFrame head, чтобы он возвращал по умолчанию 3 строки вместо 5.

from functools import partialmethod
pd.DataFrame.head = partialmethod(pd.DataFrame.head, n=3)
bikes.head()

Зачем это делать?

Я часто использую метод head, чтобы сократить отображаемый результат DataFrame. Когда я создаю интерактивные учебные пособия, 5 строк могут занимать слишком много места на экране, поэтому изменение этого значения по умолчанию на 2 или 3 имеет смысл и позволяет сэкономить немного времени. Это также может быть полезно, если вы написали большой отчет, в котором используется множество вызовов метода head, и вы хотите сократить все эти выходные данные с помощью одной команды.

Общее использование

partialmethod доступен только в Python 3 и может использоваться для всех методов, чтобы установить некоторые или все параметры на определенное значение.

Магистр Python, Data Science и машинного обучения

Погрузитесь в мой комплексный путь к овладению наукой о данных и машинным обучением с помощью Python. Купите All Access Pass, чтобы получить пожизненный доступ ко всем текущим и будущим курсам. Некоторые из курсов, которые он содержит:

Получите All Access Pass прямо сейчас!