Как использовать метод head() в pandas dataframe для получения первых 5 строк?
Как получить первые 5 строк из pandas dataframe?
Для того чтобы получить первые 5 строк из pandas dataframe, вы можете использовать метод head(5).
import pandas as pd
# Создание pandas dataframe
data = {'Имя': ['Алексей', 'Мария', 'Иван', 'Ольга', 'Дмитрий'],
'Возраст': [25, 28, 30, 24, 32],
'Город': ['Москва', 'Санкт-Петербург', 'Новосибирск', 'Екатеринбург', 'Казань']}
df = pd.DataFrame(data)
# Получение первых 5 строк
first_five_rows = df.head(5)
print(first_five_rows)
Выполнение этого кода выведет первые 5 строк из dataframe:
Имя Возраст Город
0 Алексей 25 Москва
1 Мария 28 Санкт-Петербург
2 Иван 30 Новосибирск
3 Ольга 24 Екатеринбург
4 Дмитрий 32 Казань
Детальный ответ
Как получить первые 5 строк из pandas dataframe
В pandas dataframe - это структура данных, предоставляемая библиотекой pandas для удобной и эффективной работы с данными. Иногда нам может потребоваться получить только первые несколько строк из dataframe, чтобы быстро ознакомиться с данными. В этой статье мы рассмотрим, как получить первые 5 строк из pandas dataframe.
Использование метода head()
В pandas есть метод head()
, который позволяет нам получить первые несколько строк из dataframe. Он принимает необязательный аргумент n
, который указывает, сколько строк нужно получить.
import pandas as pd
# Создаем пример dataframe
data = {'Имя': ['Анна', 'Боб', 'Карл', 'Джон', 'Эмма'],
'Возраст': [24, 32, 28, 45, 29],
'Город': ['Москва', 'Нью-Йорк', 'Лондон', 'Сидней', 'Токио']}
df = pd.DataFrame(data)
# Первые 5 строк dataframe
first_5_rows = df.head(5)
В приведенном выше примере мы сначала импортируем библиотеку pandas, затем создаем dataframe с данными о людях (имя, возраст, город). Затем мы используем метод head(5)
, чтобы получить первые 5 строк из dataframe и сохранить их в переменную first_5_rows
. Теперь first_5_rows
содержит первые 5 строк dataframe.
Результат
После выполнения примера кода переменная first_5_rows
будет содержать следующее:
Имя Возраст Город
0 Анна 24 Москва
1 Боб 32 Нью-Йорк
2 Карл 28 Лондон
3 Джон 45 Сидней
4 Эмма 29 Токио
Как видно из примера, метод head()
возвращает первые 5 строк из dataframe в виде нового dataframe с сохраненными значениями. Столбцы сохраняются в том же порядке, в котором они были в исходном dataframe.
Итоги
Итак, мы рассмотрели, как получить первые 5 строк из pandas dataframe. Мы использовали метод head()
и передали ему аргумент n=5
для получения первых 5 строк. Метод head()
возвращает новый dataframe с указанным количеством строк. Теперь вы знаете, как быстро получить первые 5 строк из dataframe в pandas.