Welcome to OGeek Q&A Community for programmer and developer-Open, Learning and Share
Welcome To Ask or Share your Answers For Others

Categories

0 votes
347 views
in Technique[技术] by (71.8m points)

python - Filter pyspark dataframe based on list of strings

I am very new to the Pyspark. I hope I get an answer here. I need an answer using DataFrame API

My question is to find the number of lines in the text file test.txt that contain words “testA” or “testB” or “testC”

lines=spark.read.text("C:est.txt")
listStr=["testA","testB","testC"]

lines.filter(lines.isin(listStr)).count()  --> this is showing all the lines in the textfile

P.S: better if it can be solved using "lambda"

question from:https://stackoverflow.com/questions/66058539/filter-pyspark-dataframe-based-on-list-of-strings

与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…
Welcome To Ask or Share your Answers For Others

1 Reply

0 votes
by (71.8m points)

If you want to use lambda function, you can use RDD:

lines.rdd.filter(lambda r: any(s in r[0] for s in listStr)).count()

与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…
OGeek|极客中国-欢迎来到极客的世界,一个免费开放的程序员编程交流平台!开放,进步,分享!让技术改变生活,让极客改变未来! Welcome to OGeek Q&A Community for programmer and developer-Open, Learning and Share
Click Here to Ask a Question

...