PySpark RDD过滤问题:如何移除含指定列表元素的行?
解决Spark RDD过滤包含指定单词行的问题
你的代码逻辑有两处关键问题,导致无法得到预期结果:
- 错误地使用了字符串
"remove.values"作为判断条件,而没有引用实际的remove列表中的单词 - 过滤逻辑没有正确检查行中的单词是否属于需要移除的列表
正确的解决方案(保留原始行文本)
如果你的目标是保留完整的原始行(和预期结果一致),不需要提前拆分单词,直接对每行文本进行判断即可:
# 读取文件,保留原始行格式 RDD = sc.textFile("myfiles/names.txt") remove = ['it','name'] # 过滤逻辑:仅保留不包含remove列表中任何单词的行 result = RDD.filter(lambda line: not any(word in line.split() for word in remove)) # 输出结果 for line in result.collect(): print(line)
如果需要先拆分单词处理(后续有其他操作)
如果你确实需要先将行拆分为单词列表,过滤后可以再还原为原始行格式:
RDD = sc.textFile("myfiles/names.txt").map(lambda x: x.split()) remove = ['it','name'] # 过滤掉包含remove列表中任意单词的行 filtered_words_rdd = RDD.filter(lambda words: not any(word in remove for word in words)) # 将单词列表还原为原始行文本 result = filtered_words_rdd.map(lambda words: ' '.join(words)) for line in result.collect(): print(line)
逻辑解释
any(word in line.split() for word in remove)会遍历remove列表中的每个单词,检查当前行拆分后的单词是否存在匹配项。如果有任何一个匹配,返回True- 加上
not后,就会保留那些没有任何匹配单词的行,正好符合你的需求
内容的提问来源于stack exchange,提问作者Abdalla Issa Mbaideen
相关产品推荐
相关产品推荐

