Assigningparenttosparkrow
创始人
2024-09-20 18:00:08
0

将Spark行分配给父行

通常在处理具有分层结构的数据时,我们会需要为Spark DataFrame中的每一行分配一个父行。这可以通过使用Spark中的Column(列)和struct(结构)类型来实现。

例如,考虑以下数据:

id parent
1 null
2 1
3 1
4 2
5 3

要将每个行Spark分配给其父行,我们可以使用withColumn函数并使用struct类型来创建一个新列。下面是例子:

from pyspark.sql.functions import struct

df = spark.createDataFrame([(1, None), (2, 1), (3, 1), (4, 2), (5, 3)], ["id", "parent"])

Assign parent to each row

df = df.withColumn("row_and_parent", struct("*", "parent"))

Show the result

df.show()

输出结果如下:

+---+------+----------------+ | id|parent| row_and_parent| +---+------+----------------+ | 1| null| [1,null] | | 2| 1| [2,1] | | 3| 1| [3,1] | | 4| 2| [4,2] | | 5| 3| [5,3] | +---+------+----------------+

现在,我们可以访问每行的父行并执行操作。例如,要过滤具有id = 4的行的父行,我们可以使用以下代码:

Filter rows with id=4 and its parent row

df.filter("id = 4 or row_and_parent.parent = 4").show()

输出结果:

+---+------+----------------+ | id|parent| row_and_parent| +---+------+----------------+ | 2|

相关内容

热门资讯

透视智能ai!哈糖大菠萝怎么开... 透视智能ai!哈糖大菠萝怎么开挂(透视)必备教程(2022已更新)(哔哩哔哩)1、哈糖大菠萝怎么开挂...
透视智能ai!aapoker透... 透视智能ai!aapoker透视方法(透视)脚本(果然真的是有挂);1、不需要AI权限,帮助你快速的...
透视黑科技!佛手大菠萝13道挂... 透视黑科技!佛手大菠萝13道挂哪里(透视)2025版教程(2020已更新)(哔哩哔哩)佛手大菠萝13...
透视辅助!pokemmo辅助脚... 透视辅助!pokemmo辅助脚本(透视)教你攻略(2020已更新)(哔哩哔哩)1、不需要AI权限,帮...
透视实锤!aapoker辅助工... 透视实锤!aapoker辅助工具安全吗(透视)发牌逻辑(原来存在有挂);1、透视实锤!aapoker...
透视辅助!智星菠萝辅助(透视)... 透视辅助!智星菠萝辅助(透视)必备教程(2025已更新)(哔哩哔哩)1、上手简单,内置详细流程视频教...
透视规律!aapoker插件下... 透视规律!aapoker插件下载(透视)插件(总是真的有挂)aapoker插件下载辅助器中分为三种模...
透视ai!哈糖大菠萝助手(透视... 透视ai!哈糖大菠萝助手(透视)技巧教程(2026已更新)(哔哩哔哩);1、全新机制【哈糖大菠萝助手...
透视辅助!aa poker透视... 透视辅助!aa poker透视软件(透视)透视软件(好像存在有挂);aa poker透视软件辅助器中...
透视代打!agpoker辅助(... 透视代打!agpoker辅助(透视)wpk教程(2023已更新)(哔哩哔哩)1、任何agpoker辅...