我想收集数组列表中的数据框列值来进行一些计算，这可能吗？

Question

我正在通过这个从 phoenix 加载数据：

val tableDF = sqlContext.phoenixTableAsDataFrame("Hbtable", Array("ID", "distance"), conf = configuration)

并希望对列值 distance 执行以下计算：

val list=Array(10,20,30,40,10,20,0,10,20,30,40,50,60)//list of values from the column distance
val first=list(0)
val last=list(list.length-1)
var m = 0; 
for (a <- 0 to list.length-2) {
  if (list(a + 1) < list(a) && list(a+1)>=0)
  {
     m = m + list(a)
  } 
}
val totalDist=(m+last-first)

Answer 1

你可以这样做。它returnsArray[Any]

`val array = df.select("distance").rdd.map(r => r(0)).collect()

如果你想正确获取数据类型，那么你可以使用。它 returns Array[Int]

val array = df.select("distance").rdd.map(r => r(0).asInstanceOf[Int]).collect()