как получить элемент фрейма данных искры? - PullRequest
0 голосов
/ 09 июля 2020

Предположим, у меня есть фрейм данных, как показано ниже:

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

Как мне получить значение c, где a = 1 и b = 2?

Ответы [ 2 ]

0 голосов
/ 10 июля 2020

val col_c = df.filter(col("a").equalTo(1).and(col("b").equalTo(2))).select("c")

0 голосов
/ 10 июля 2020
val resDF = df.filter(col("a").equalTo(1)).filter(col("b").equalTo(2))
...