按照指定列进行分组并在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方法显示结果数据帧。

相关内容

热门资讯

透视能赢(德州微扑克专用)外挂... 透视能赢(德州微扑克专用)外挂透明挂辅助器安装(辅助挂)透视辅助(2025已更新)(哔哩哔哩);亲,...
发现一款(哈糖大菠萝平台)外挂... 发现一款(哈糖大菠萝平台)外挂透明挂辅助软件(透视)原来是真的有挂(可靠教程)(哔哩哔哩)是一款可以...
透视存在(wpk测试)外挂透明... 透视存在(wpk测试)外挂透明挂辅助神器(辅助挂)辅助透视(2020已更新)(哔哩哔哩);亲们利用一...
透视好友房(WPK开挂)外挂透... 透视好友房(WPK开挂)外挂透明挂辅助挂(辅助挂)原来真的有挂(切实教程)(哔哩哔哩),亲,有的,a...
专业讨论(aapoker手游版... 专业讨论(aapoker手游版)外挂透明挂辅助挂(透视)软件透明挂(2022已更新)(哔哩哔哩);值...
透视游戏(德扑之星机制)外挂透... 透视游戏(德扑之星机制)外挂透明挂辅助APP(透视)原来真的有挂(必胜教程)(哔哩哔哩);wpk透视...
分享实测(wePoke)外挂透... 分享实测(wePoke)外挂透明挂辅助工具(透视)软件透明挂(2021已更新)(哔哩哔哩)1、玩家可...
玩家必看科普(德州透视)外挂透... 玩家必看科普(德州透视)外挂透明挂辅助器安装(透视)透视辅助(确实有挂)-哔哩哔哩;wpk透视辅助官...
交流学习经验(鱼扑克app a... 交流学习经验(鱼扑克app ai)外挂透明挂辅助脚本(透视)其实是真的有挂(安装教程)(哔哩哔哩);...
技术分享(wepoke ai)... 技术分享(wepoke ai)外挂透明挂辅助器(透视)软件透明挂(2023已更新)(哔哩哔哩)关于w...