如何根据列中包含的值过滤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

在使用k近邻算法时,有没有办法获取被使用的“邻居”?

我想找到一种方法来确定在我的knn算法中实际使用了哪些…

Theano在Google Colab上无法启用GPU支持

我在尝试使用Theano库训练一个模型。由于我的电脑内…

准确性评分似乎有误

这里是代码: from sklearn.metrics…

Keras Functional API: “错误检查输入时:期望input_1具有4个维度,但得到形状为(X, Y)的数组”

我在尝试使用Keras的fit_generator来训…

如何使用sklearn.datasets.make_classification在指定范围内生成合成数据?

我想为分类问题创建合成数据。我使用了sklearn.d…

如何处理预测时不在训练集中的标签

已关闭。 此问题与编程或软件开发无关。目前不接受回答。…

发表回复

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