在PySpark中利用数组列与列表交集进行DataFrame过滤的正确姿势

在PySpark中利用数组列与列表交集进行DataFrame过滤的正确姿势

本文详细介绍了如何在PySpark中高效地过滤DataFrame,当需要根据数组列与一个给定Python列表的交集来筛选数据时。核心解决方案是利用pyspark.sql.functions.arrays_overlap函数,并结合lit函数将Python列表中的元素转换为Spark字面量表达式,从而构建正确的过滤条件,避免常见的AnalysisException错误。文章提供了清晰的示例代码和关键概念解释,旨在帮助用户正确实现此类复杂过滤逻辑。

1. 问题背景与挑战

在数据处理中,我们经常需要对包含数组类型列的spark dataframe进行过滤。一个常见的需求是,筛选出那些数组列中至少包含给定python列表(例如 [item1, item2, …])中一个或多个元素的行。

在SQL中,这种操作非常直观,通常可以使用arrays_overlap函数:

SELECT FROM WHERE arrays_overlap(, array())

然而,当尝试将这种逻辑直接转换为PySpark时,许多用户会遇到困难。一个常见的错误尝试是:

from pyspark.sql.functions import col, array, arrays_overlap# 假设 target_list 是一个 Python 列表,如 ['apple', 'banana']df.filter(arrays_overlap(col("array_column"), array(target_list)))

这段代码通常会导致AnalysisException,错误信息类似于[UNRESOLVED_COLUMN.WITH_SUGGESTION] A column or function parameter with name ” cannot be resolved.。这是因为array()函数在接收非列参数时,期望的是字面量表达式(literal expressions),而不是原始的Python列表元素。虽然array_contains函数可以处理单个元素,但它无法满足与整个列表进行交集判断的需求。

2. 解决方案:结合 lit 函数

解决这个问题的关键在于,将Python列表中的每个元素转换为Spark的字面量表达式(literal expression),然后再用array函数将其组合成一个字面量数组。这可以通过pyspark.sql.functions.lit函数来实现。

lit函数的作用是将一个Python值转换为一个Spark列表达式,这个表达式代表着一个常量值。当我们将列表中的每个元素都通过lit转换后,再将这些字面量表达式传递给array函数,array函数就能正确地构建一个包含这些字面量值的数组。

正确的PySpark实现如下:

from pyspark.sql import SparkSessionfrom pyspark.sql.functions import col, array, arrays_overlap, lit# 1. 初始化 SparkSessionspark = SparkSession.builder.appName("ArrayColumnFilter").getOrCreate()# 2. 准备示例数据data = [    (1, ["apple", "banana", "orange"]),    (2, ["grape", "kiwi"]),    (3, ["banana", "strawberry"]),    (4, ["mango", "pineapple"]),    (5, ["apple", "grape"])]df = spark.createDataFrame(data, ["id", "fruits_array"])df.printSchema()df.show()# 3. 定义用于过滤的 Python 列表target_list = ["banana", "grape", "lemon"]# 4. 构建正确的过滤条件# 使用 map(lit, target_list) 将列表中的每个元素转换为 lit 表达式# 使用 * 解包这些 lit 表达式作为 array 函数的参数# 最后,使用 arrays_overlap 进行比较filtered_df = df.filter(    arrays_overlap(col("fruits_array"), array(*map(lit, target_list))))# 5. 显示过滤结果print(f"n原始DataFrame:")df.show()print(f"n过滤列表:{target_list}")print("n过滤后的DataFrame(fruits_array与target_list有交集):")filtered_df.show()# 6. 停止 SparkSessionspark.stop()

运行结果示例:

root |-- id: long (nullable = true) |-- fruits_array: array (nullable = true) |    |-- element: string (nullable = true)+---+--------------------+| id|        fruits_array|+---+--------------------+|  1|[apple, banana, o...||  2|       [grape, kiwi]||  3|[banana, strawber...||  4|[mango, pineapple]||  5|       [apple, grape]|+---+--------------------+原始DataFrame:+---+--------------------+| id|        fruits_array|+---+--------------------+|  1|[apple, banana, o...||  2|       [grape, kiwi]||  3|[banana, strawber...||  4|[mango, pineapple]||  5|       [apple, grape]|+---+--------------------+过滤列表:['banana', 'grape', 'lemon']过滤后的DataFrame(fruits_array与target_list有交集):+---+--------------------+| id|        fruits_array|+---+--------------------+|  1|[apple, banana, o...||  2|       [grape, kiwi]||  3|[banana, strawber...||  5|       [apple, grape]|+---+--------------------+

从结果可以看出,id为1、2、3、5的行被保留,因为它们的fruits_array列与[“banana”, “grape”, “lemon”]存在交集(例如,id=1包含”banana”,id=2包含”grape”,id=3包含”banana”,id=5包含”grape”)。

3. 关键概念与注意事项

arrays_overlap(array1, array2): 这个函数用于判断两个数组是否有共同的元素。如果存在任何共同元素,则返回True;否则返回False。它是进行数组交集判断的核心。lit(value): lit函数将一个Python字面量(如字符串、数字、布尔值)转换为一个Spark SQL的字面量列。这是在PySpark中构建常量值表达式的常用方法。*`array(expressions)**:array`函数有两种主要用法:当参数是列名时,它将这些列的值组合成一个新的数组列。例如:array(col(“col1”), col(“col2”))。当参数是字面量表达式时,它会创建一个包含这些字面量值的字面量数组。例如:array(lit(“a”), lit(“b”))。我们这里的解决方案属于第二种情况,map(lit, target_list)生成了一系列字面量表达式,*操作符将它们解包作为array函数的独立参数。错误避免: 理解array函数对参数类型的期望是避免AnalysisException的关键。直接传递Python列表array(target_list)会被Spark误解为target_list中的第一个元素是一个列名,因此无法解析。

4. 总结

在PySpark中,当需要使用一个Python列表与DataFrame的数组列进行交集过滤时,务必记住使用pyspark.sql.functions.lit函数将列表中的每个元素转换为Spark字面量表达式。然后,通过array(*map(lit, your_list))的方式构建一个字面量数组,并将其作为arrays_overlap函数的第二个参数。这种模式是处理这类复杂数组过滤逻辑的标准且正确的方法,能够确保代码的健壮性和准确性。

以上就是在PySpark中利用数组列与列表交集进行DataFrame过滤的正确姿势的详细内容,更多请关注创想鸟其它相关文章!

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/1373239.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2025年12月14日 13:04:57
下一篇 2025年12月14日 13:05:11

相关推荐

  • Uniapp 中如何不拉伸不裁剪地展示图片?

    灵活展示图片:如何不拉伸不裁剪 在界面设计中,常常需要以原尺寸展示用户上传的图片。本文将介绍一种在 uniapp 框架中实现该功能的简单方法。 对于不同尺寸的图片,可以采用以下处理方式: 极端宽高比:撑满屏幕宽度或高度,再等比缩放居中。非极端宽高比:居中显示,若能撑满则撑满。 然而,如果需要不拉伸不…

    2025年12月24日
    400
  • 如何让小说网站控制台显示乱码,同时网页内容正常显示?

    如何在不影响用户界面的情况下实现控制台乱码? 当在小说网站上下载小说时,大家可能会遇到一个问题:网站上的文本在网页内正常显示,但是在控制台中却是乱码。如何实现此类操作,从而在不影响用户界面(UI)的情况下保持控制台乱码呢? 答案在于使用自定义字体。网站可以通过在服务器端配置自定义字体,并通过在客户端…

    2025年12月24日
    800
  • 如何在地图上轻松创建气泡信息框?

    地图上气泡信息框的巧妙生成 地图上气泡信息框是一种常用的交互功能,它简便易用,能够为用户提供额外信息。本文将探讨如何借助地图库的功能轻松创建这一功能。 利用地图库的原生功能 大多数地图库,如高德地图,都提供了现成的信息窗体和右键菜单功能。这些功能可以通过以下途径实现: 高德地图 JS API 参考文…

    2025年12月24日
    400
  • 如何使用 scroll-behavior 属性实现元素scrollLeft变化时的平滑动画?

    如何实现元素scrollleft变化时的平滑动画效果? 在许多网页应用中,滚动容器的水平滚动条(scrollleft)需要频繁使用。为了让滚动动作更加自然,你希望给scrollleft的变化添加动画效果。 解决方案:scroll-behavior 属性 要实现scrollleft变化时的平滑动画效果…

    2025年12月24日
    000
  • 如何为滚动元素添加平滑过渡,使滚动条滑动时更自然流畅?

    给滚动元素平滑过渡 如何在滚动条属性(scrollleft)发生改变时为元素添加平滑的过渡效果? 解决方案:scroll-behavior 属性 为滚动容器设置 scroll-behavior 属性可以实现平滑滚动。 html 代码: click the button to slide right!…

    2025年12月24日
    500
  • 如何选择元素个数不固定的指定类名子元素?

    灵活选择元素个数不固定的指定类名子元素 在网页布局中,有时需要选择特定类名的子元素,但这些元素的数量并不固定。例如,下面这段 html 代码中,activebar 和 item 元素的数量均不固定: *n *n 如果需要选择第一个 item元素,可以使用 css 选择器 :nth-child()。该…

    2025年12月24日
    200
  • 使用 SVG 如何实现自定义宽度、间距和半径的虚线边框?

    使用 svg 实现自定义虚线边框 如何实现一个具有自定义宽度、间距和半径的虚线边框是一个常见的前端开发问题。传统的解决方案通常涉及使用 border-image 引入切片图片,但是这种方法存在引入外部资源、性能低下的缺点。 为了避免上述问题,可以使用 svg(可缩放矢量图形)来创建纯代码实现。一种方…

    2025年12月24日
    100
  • 如何解决本地图片在使用 mask JS 库时出现的跨域错误?

    如何跨越localhost使用本地图片? 问题: 在本地使用mask js库时,引入本地图片会报跨域错误。 解决方案: 要解决此问题,需要使用本地服务器启动文件,以http或https协议访问图片,而不是使用file://协议。例如: python -m http.server 8000 然后,可以…

    2025年12月24日
    200
  • 如何让“元素跟随文本高度,而不是撑高父容器?

    如何让 元素跟随文本高度,而不是撑高父容器 在页面布局中,经常遇到父容器高度被子元素撑开的问题。在图例所示的案例中,父容器被较高的图片撑开,而文本的高度没有被考虑。本问答将提供纯css解决方案,让图片跟随文本高度,确保父容器的高度不会被图片影响。 解决方法 为了解决这个问题,需要将图片从文档流中脱离…

    2025年12月24日
    000
  • 为什么 CSS mask 属性未请求指定图片?

    解决 css mask 属性未请求图片的问题 在使用 css mask 属性时,指定了图片地址,但网络面板显示未请求获取该图片,这可能是由于浏览器兼容性问题造成的。 问题 如下代码所示: 立即学习“前端免费学习笔记(深入)”; icon [data-icon=”cloud”] { –icon-cl…

    2025年12月24日
    200
  • 如何利用 CSS 选中激活标签并影响相邻元素的样式?

    如何利用 css 选中激活标签并影响相邻元素? 为了实现激活标签影响相邻元素的样式需求,可以通过 :has 选择器来实现。以下是如何具体操作: 对于激活标签相邻后的元素,可以在 css 中使用以下代码进行设置: li:has(+li.active) { border-radius: 0 0 10px…

    2025年12月24日
    100
  • 如何模拟Windows 10 设置界面中的鼠标悬浮放大效果?

    win10设置界面的鼠标移动显示周边的样式(探照灯效果)的实现方式 在windows设置界面的鼠标悬浮效果中,光标周围会显示一个放大区域。在前端开发中,可以通过多种方式实现类似的效果。 使用css 使用css的transform和box-shadow属性。通过将transform: scale(1.…

    2025年12月24日
    200
  • 为什么我的 Safari 自定义样式表在百度页面上失效了?

    为什么在 Safari 中自定义样式表未能正常工作? 在 Safari 的偏好设置中设置自定义样式表后,您对其进行测试却发现效果不同。在您自己的网页中,样式有效,而在百度页面中却失效。 造成这种情况的原因是,第一个访问的项目使用了文件协议,可以访问本地目录中的图片文件。而第二个访问的百度使用了 ht…

    2025年12月24日
    000
  • 如何用前端实现 Windows 10 设置界面的鼠标移动探照灯效果?

    如何在前端实现 Windows 10 设置界面中的鼠标移动探照灯效果 想要在前端开发中实现 Windows 10 设置界面中类似的鼠标移动探照灯效果,可以通过以下途径: CSS 解决方案 DEMO 1: Windows 10 网格悬停效果:https://codepen.io/tr4553r7/pe…

    2025年12月24日
    000
  • 使用CSS mask属性指定图片URL时,为什么浏览器无法加载图片?

    css mask属性未能加载图片的解决方法 使用css mask属性指定图片url时,如示例中所示: mask: url(“https://api.iconify.design/mdi:apple-icloud.svg”) center / contain no-repeat; 但是,在网络面板中却…

    2025年12月24日
    000
  • 如何用CSS Paint API为网页元素添加时尚的斑马线边框?

    为元素添加时尚的斑马线边框 在网页设计中,有时我们需要添加时尚的边框来提升元素的视觉效果。其中,斑马线边框是一种既醒目又别致的设计元素。 实现斜向斑马线边框 要实现斜向斑马线间隔圆环,我们可以使用css paint api。该api提供了强大的功能,可以让我们在元素上绘制复杂的图形。 立即学习“前端…

    2025年12月24日
    000
  • 图片如何不撑高父容器?

    如何让图片不撑高父容器? 当父容器包含不同高度的子元素时,父容器的高度通常会被最高元素撑开。如果你希望父容器的高度由文本内容撑开,避免图片对其产生影响,可以通过以下 css 解决方法: 绝对定位元素: .child-image { position: absolute; top: 0; left: …

    2025年12月24日
    000
  • 使用 Mask 导入本地图片时,如何解决跨域问题?

    跨域疑难:如何解决 mask 引入本地图片产生的跨域问题? 在使用 mask 导入本地图片时,你可能会遇到令人沮丧的跨域错误。为什么会出现跨域问题呢?让我们深入了解一下: mask 框架假设你以 http(s) 协议加载你的 html 文件,而当使用 file:// 协议打开本地文件时,就会产生跨域…

    2025年12月24日
    200
  • CSS 帮助

    我正在尝试将文本附加到棕色框的左侧。我不能。我不知道代码有什么问题。请帮助我。 css .hero { position: relative; bottom: 80px; display: flex; justify-content: left; align-items: start; color:…

    2025年12月24日 好文分享
    200
  • 前端代码辅助工具:如何选择最可靠的AI工具?

    前端代码辅助工具:可靠性探讨 对于前端工程师来说,在HTML、CSS和JavaScript开发中借助AI工具是司空见惯的事情。然而,并非所有工具都能提供同等的可靠性。 个性化需求 关于哪个AI工具最可靠,这个问题没有一刀切的答案。每个人的使用习惯和项目需求各不相同。以下是一些影响选择的重要因素: 立…

    2025年12月24日
    000

发表回复

登录后才能评论
关注微信