spark dataframe 转list 实体类
你可以使用collect()函数将Spark DataFrame转换为Python List,并使用Row类将每一行转换为实体类对象。以下是示例代码:
from pyspark.sql import Row
# 假设你的DataFrame名为df,包含两列name和age
# 将DataFrame转换为Python List
df_list = df.collect()
# 定义实体类
class Person:
def __init__(self, name, age):
self.name = name
self.age = age
# 将List中的每一行转换为Person对象
person_list = [Person(row.name, row.age) for row in df_list]
现在,你可以使用person_list来操作实体类对象。
原文地址: https://www.cveoy.top/t/topic/ZGt 著作权归作者所有。请勿转载和采集!