format (DataFrameReader)

指定输入数据源格式。

Syntax

format(source)

参数

参数 类型 说明
source str 数据源的名称,例如 'json''parquet'

退货

DataFrameReader

示例

将 DataFrame 写入 JSON 文件,并将其读回。

import tempfile
with tempfile.TemporaryDirectory(prefix="format") as d:
    spark.createDataFrame(
        [{"age": 100, "name": "Alice"}]
    ).write.mode("overwrite").format("json").save(d)

    spark.read.format('json').load(d).show()
    # +---+------------+
    # |age|        name|
    # +---+------------+
    # |100|Alice|
    # +---+------------+