합계 어레이 - 스칼라

Question

는 I 스칼라에서 벡터의 어레이를 가지고

import org.apache.mahout.math.{ VectorWritable, Vector, DenseVector } 
import org.apache.mahout.clustering.dirichlet.UncommonDistributions 

    val data = new ArrayBuffer[Vector]() 
    for (i <- 100 to num) { 
     data += new DenseVector(Array[Double](

     i % 30, 

     UncommonDistributions.rNorm(100, 100), 

     UncommonDistributions.rNorm(100, 100) 
    ) 



} 

제가 첫번째 행에 의해 그룹화하는 두 번째 및 세 번째 필드를 합산한다고 가정하자.

더 좋은 방법은 무엇입니까?

Answer 1

사용 GROUPBY 기능, 각 그룹을지도 - 단순히 한 줄의 코드에서 :

data groupBy (_(0)) map { case (k,v) => k -> (v map (_(2)) sum) } 

Answer 2

나는 컬렉션에서 GROUPBY 방법 선물을 사용하는 것이 좋습니다 것입니다 : 이것은 사용자가 지정한 판별에지도 벡터의 기반을 만들 것입니다

http://www.scala-lang.org/api/current/index.html#scala.collection.immutable.Vector@groupBy[K](f:A=>K):scala.collection.immutable.Map[K,Repr]

.

편집 : 일부 코드 예제 : 목록

// I created a different Array of Vector as I don't have Mahout dependencies 
// But the output is similar 
// A List of Vectors with 3 values inside 
val num = 100 
val data = (0 to num).toList.map(n => { 
    Vector(n % 30, n/100, n * 100) 
}) 

// The groupBy will create a Map of Vectors where the Key is the result of the function 
// And here, the function return the first value of the Vector 
val group = data.groupBy(v => { v.apply(0) }) 

// Also a subset of the result: 
// group: 
// scala.collection.immutable.Map[Int,List[scala.collection.immutable.Vector[Int]]] = Map(0 -> List(Vector(0, 0, 0), Vector(0, 0, 3000), Vector(0, 0, 6000), Vector(0, 0, 9000)), 5 -> List(Vector(5, 0, 500), Vector(5, 0, 3500), Vector(5, 0, 6500), Vector(5, 0, 9500)))