按照指定列进行分组并在PySpark数据帧中创建新列
创始人
2024-08-26 07:30:05
0

在PySpark中,可以使用groupBywithColumn方法来按照指定列进行分组并创建新列。

以下是一个示例代码:

# 导入必要的模块
from pyspark.sql import SparkSession
from pyspark.sql.functions import col

# 创建SparkSession对象
spark = SparkSession.builder.getOrCreate()

# 创建示例数据帧
data = [("Alice", 25, "A"),
        ("Bob", 30, "B"),
        ("Alice", 35, "A"),
        ("Bob", 40, "B")]

df = spark.createDataFrame(data, ["Name", "Age", "Group"])

# 按照指定列进行分组,并创建新列"Average Age"来计算每个组的平均年龄
df_with_avg_age = df.groupBy("Group").agg({"Age": "avg"}).withColumnRenamed("avg(Age)", "Average Age")

# 显示结果
df_with_avg_age.show()

运行以上代码,将会得到以下输出:

+-----+-----------+
|Group|Average Age|
+-----+-----------+
|    B|       35.0|
|    A|       30.0|
+-----+-----------+

在上面的代码中,我们使用groupBy方法按照"Group"列进行分组,并使用agg方法计算每个组的平均年龄。然后,使用withColumnRenamed方法为结果列重命名为"Average Age"。最后,使用show方法显示结果数据帧。

相关内容

热门资讯

黑科技辅助!wepoke软件靠... 您好,wepoke软件靠谱这款游戏可以开挂的,确实是有挂的,需要了解加微【136704302】很多玩...
黑科技辅助!wepoke插件,... 黑科技辅助!wepoke插件,(WePoKe有规律)原来真的有挂,辅助教程(有挂透视)-哔哩哔哩是一...
黑科技辅助!wepoke黑科技... 黑科技辅助!wepoke黑科技是啥,(wepoke德州扑克)原来真的有挂,详细教程(有挂分享)-哔哩...
一分钟了解!gg扑克有假,(扑... 一分钟了解!gg扑克有假,(扑克时间)原来真的有挂,软件教程(了解有挂)-哔哩哔哩;一分钟了解!gg...
黑科技辅助!wepoke私人局... 您好,wepoke私人局有挂这款游戏可以开挂的,确实是有挂的,需要了解加微【136704302】很多...
科普攻略!微扑克辅牌器,(新微... 科普攻略!微扑克辅牌器,(新微扑克)原来真的有挂,可靠技巧(有挂解惑)-哔哩哔哩是一款可以让一直输的...
黑科技辅助!wepoke有辅助... 黑科技辅助!wepoke有辅助软件,(WePoKe能胜)原来真的有挂,必胜教程(有挂详情)-哔哩哔哩...
黑科技辅助!wepoke软件透... 黑科技辅助!wepoke软件透明挂合法,(WePoKe胜率)原来真的有挂,系统教程(有挂规律)-哔哩...
黑科技辅助!WePoKe透明挂... 您好:WePoKe透明挂这款游戏可以开挂的,确实是有挂的,很多玩家在这款游戏中打牌都会发现很多用户的...
分享个大家!云扑克是否有外挂,... 分享个大家!云扑克是否有外挂,(云扑克软件)原来真的有挂,新2025版(真实有挂)-哔哩哔哩是一款可...