您可以使用 collect() 方法将 Spark DataFrame 转换为 Python 列表。例如:

df = spark.createDataFrame([(1, 'John'), (2, 'Jane'), (3, 'Bob')], ['id', 'name'])
data_list = df.collect()
print(data_list)

输出:

[Row(id=1, name='John'), Row(id=2, name='Jane'), Row(id=3, name='Bob')]

请注意,如果您的 DataFrame 非常大,则不建议使用 collect() 方法,因为它将所有数据收集到驱动程序中,这可能会导致内存问题。相反,您可以使用其他 Spark 操作来处理数据,例如 filter()groupBy()agg() 等。

Spark DataFrame 如何转换为 Python 列表

原文地址: https://www.cveoy.top/t/topic/mj2E 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录