如何根据列中包含的值过滤Spark Dataframe?

在我的数据集中,某些字符串列包含None或NA值,我想统计这些空值的数量,以便根据统计结果决定如何处理缺失值。

我尝试了一种效率不高的过滤方法,即对每个列使用或表达式进行过滤。我希望找到一种更高效和简洁的方法来过滤这些值。最好不要将数据转换为RDD,但如果必须使用RDD来实现这种过滤方式也是可以接受的。

我找到了这个讨论 Spark SQL filter multiple fields,与我的问题相似,但由于我有很多列,我希望找到一种更简洁和优雅的方式来编写代码。

// trainDataFull是我的数据框架val nullValues = Array("NA", "None")val filtered = trainDataFull.filter(trainDataFull("Alley").isin(nullValues:_*) ||      trainDataFull("MSZoning").isin(nullValues:_*) ||      trainDataFull("Street").isin(nullValues:_*) ||      trainDataFull("LotShape").isin(nullValues:_*) ||      trainDataFull("LandContour").isin(nullValues:_*) ||      trainDataFull("Utilities").isin(nullValues:_*) ||      trainDataFull("LotConfig").isin(nullValues:_*) ||      trainDataFull("LandSlope").isin(nullValues:_*) ||      trainDataFull("Neighborhood").isin(nullValues:_*) ||      trainDataFull("Condition1").isin(nullValues:_*) ||      trainDataFull("Condition2").isin(nullValues:_*) ||      trainDataFull("BldgType").isin(nullValues:_*) ||      trainDataFull("HouseStyle").isin(nullValues:_*) ||      trainDataFull("RoofStyle").isin(nullValues:_*) ||      trainDataFull("RoofMatl").isin(nullValues:_*) ||      trainDataFull("Exterior1st").isin(nullValues:_*) ||      trainDataFull("Exterior2nd").isin(nullValues:_*) ||      trainDataFull("MasVnrType").isin(nullValues:_*) ||      trainDataFull("MasVnrArea").isin(nullValues:_*) ||      trainDataFull("ExterQual").isin(nullValues:_*) ||      trainDataFull("MasVnrArea").isin(nullValues:_*) ||      trainDataFull("ExterQual").isin(nullValues:_*) ||      trainDataFull("ExterCond").isin(nullValues:_*) ||      trainDataFull("Foundation").isin(nullValues:_*) ||      trainDataFull("BsmtQual").isin(nullValues:_*) ||      trainDataFull("BsmtCond").isin(nullValues:_*) ||      trainDataFull("BsmtExposure").isin(nullValues:_*)    )

我想查看每个列中包含多少个空值。


回答:

您可以始终以编程方式生成查询

val nullValues = Array("NA", "None")val df = Seq(("NA", "Foo"), ("None", "NA")).toDF("MSZoning", "Street")val columns = df.schema.collect {  case StructField(name, StringType, _, _) =>    sum(when(col(name).isInCollection(nullValues), 1)).as(name)}df.select(columns:_*).show()

输出:

+--------+------+|MSZoning|Street|+--------+------+|       2|     1|+--------+------+

Related Posts

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注