head

Возвращает первые n строки.

Синтаксис

head(n: Optional[int] = None)

Параметры

Параметр Тип Описание
n int, необязательный значение по умолчанию 1. Количество возвращаемых строк.

Возвраты

Если задано значение n, верните список Row длины n или меньше, если кадр данных имеет меньше элементов. Если значение n отсутствует, верните одну строку.

Примечания

Этот метод следует использовать только в том случае, если результирующий массив должен быть небольшим, так как все данные загружаются в память драйвера.

Примеры

df = spark.createDataFrame([
    (2, "Alice"), (5, "Bob")], schema=["age", "name"])
df.head()
# Row(age=2, name='Alice')
df.head(1)
# [Row(age=2, name='Alice')]
df.head(0)
# []