按照id分组并根据优先级创建列(Pyspark)
创始人
2024-08-24 06:00:41
0

可以使用groupby和pivot方法来解决此问题。首先,我们需要使用groupby方法按照id列分组。然后,我们可以使用pivot方法并指定“priority”列来创建新列。最后,我们需要使用fillna方法来使用默认值填充缺失的单元格。

以下是用于解决此问题的示例代码:

from pyspark.sql.functions import *
from pyspark.sql import SparkSession

# 创建Spark会话
spark = SparkSession.builder.appName("groupby_pivot_demo").getOrCreate()
spark.conf.set("spark.sql.shuffle.partitions", "5")

# 创建示例数据
data = [("id1", "A", 1),
       ("id1", "B", 2),
       ("id1", "C", 3),
       ("id2", "A", 2),
       ("id2", "B", 1),
       ("id3", "A", 3)]

df = spark.createDataFrame(data, ["id", "priority", "value"])

# 按照id分组并创建新列
df_grouped = df.groupBy("id").pivot("priority").agg(first(col("value")))

# 填充缺失单元格
df_grouped = df_grouped.fillna(0)

df_grouped.show()

输出结果:

+---+---+---+---+
| id|  A|  B|  C|
+---+---+---+---+
|id2|  2|  1|  0|
|id1|  1|  2|  3|
|id3|  3|  0|  0|
+---+---+---+---+

相关内容

热门资讯

透视能赢!aapoker辅助怎... 透视能赢!aapoker辅助怎么用(透视)真的假的(其实是有挂);1、aapoker辅助怎么用ai辅...
透视ai代打!德普软件(透视)... 透视ai代打!德普软件(透视)app安卓版破解版(一直真的是有挂)1、下载好德普软件辅助软件之后点击...
透视讲解!aapoker怎么设... 透视讲解!aapoker怎么设置提高好牌几率(透视)脚本(好像真的是有挂);1、进入到aapoker...
透视线上!德普之星透视辅助软件... 透视线上!德普之星透视辅助软件下载(透视)免费透视(本来是有挂)1、德普之星透视辅助软件下载ai辅助...
透视美元局!aapoker公共... 透视美元局!aapoker公共底牌(透视)透视脚本下载(本来存在有挂);aapoker公共底牌辅助器...
透视ai!德普之星私人局透视(... 透视ai!德普之星私人局透视(透视)辅助器app(其实真的有挂)1、在德普之星私人局透视ai机器人技...
透视计算!aapoker公共底... 透视计算!aapoker公共底牌(透视)辅助插件工具(果然是真的有挂)1、aapoker公共底牌ai...
透视攻略!德普之星透视辅助软件... 透视攻略!德普之星透视辅助软件是真的吗(透视)透视(总是是有挂)1、用户打开应用后不用登录就可以直接...
透视脚本!aapoker透视怎... 透视脚本!aapoker透视怎么用(透视)透视脚本入口(总是是有挂)所有人都在同一条线上,像星星一样...
透视了解!德扑圈透视(透视)私... 透视了解!德扑圈透视(透视)私人局透视(确实真的是有挂);1、点击下载安装,德扑圈透视插件透视分类一...