scala获取dataframe某个值
时间: 2023-10-05 22:06:39 浏览: 85
你可以使用DataFrame的`select`和`collect`方法来获取DataFrame中的某个值。下面是一个使用Scala获取DataFrame中值的示例:
```
import org.apache.spark.sql.SparkSession
object Main {
def main(args: Array[String]): Unit = {
val spark = SparkSession
.builder()
.appName("ScalaDataFrameExample")
.master("local")
.getOrCreate()
import spark.implicits._
val data = Seq(
(1, "John"),
(2, "Jane"),
(3, "Joe")
)
val df = data.toDF("id", "name")
val name = df.select("name").where("id = 2").collect()(0)(0)
println(name)
}
}
```
这个示例中,我们先创建了一个包含id和name两列的DataFrame,然后使用`select`方法选择了名为"name"的列,并使用`where`方法过滤了id等于2的行。最后,我们使用`collect`方法将结果收集到本地,并通过索引获取第一个元素的第一个值,即"name"列中id等于2的行的值。
阅读全文