首页 > 解决方案 > 我如何获得火花数据框的元素?

问题描述

假设我有一个如下数据框:

df :
+-----------------------+-------------------------+---------------------+
| a                     | b                       | c                   |
+-----------------------+-------------------------+---------------------+
| 1                     | 1                       | 0.2                 |
| 1                     | 2                       | 0.3                 |
| 1                     | 3                       | 0.4                 |
| 1                     | 4                       | 0.5                 |
| 1                     | 5                       | 0.2                 |
+-----------------------+-------------------------+---------------------+

如何获得 a = 1 且 b = 2 的 c 值?

标签: apache-spark-sql

解决方案


val resDF = df.filter(col("a").equalTo(1)).filter(col("b").equalTo(2))

推荐阅读