take

Возвращает первые num строки в виде списка Row.

Синтаксис

take(num: int)

Параметры

Параметр Тип Описание
num int Количество возвращаемых записей. Возвращает это число записей или все записи, если кадр данных содержит меньше этого числа записей.

Возвраты

список: список строк.

Примеры

df = spark.createDataFrame(
    [(14, "Tom"), (23, "Alice"), (16, "Bob")], ["age", "name"])

df.take(2)
# [Row(age=14, name='Tom'), Row(age=23, name='Alice')]