Scala - подсчитать количество вхождений каждого ключа в Iterator

Question

У меня есть итератор, содержащий несколько пар значений ключа. например

(жень, хуг) (Ken, ZXY) (жень, ASD) (Ken, ASDF)

Результаты должны быть

(жень, 2) (ken, 2)

Как использовать функцию count (или любую другую) для подсчета количества вхождений каждой клавиши в итераторе этой конкретной коллекции.

Edit: Коллекция что итератор represend в моем потребительной случае имеет большое количество записей, возможно в диапазоне миллионов, нет мне нужен самый эффективный (меньше времени сложности) способ сделать это. Я узнал, что метод count был довольно быстрым и что его можно каким-то образом использовать для получения результата желания.

Answer 1

Вы можете groupBy ключ и затем подсчета вхождений каждого ключа:

val iterator = 
    Iterator(("jen","xyz"), ("ken","zxy"), ("jen","asd"), ("ken", "asdf")) 

iterator.toList.groupBy(_._1).mapValues(_.length).toList 
// List[(String, Int)] = List((jen,2), (ken,2)) 

Answer 2

Подход, который Питер Neyens предлагает будет работать, но это может быть очень неэффективным (время и память) для некоторых приложений из-за путь toList, groupBy и length. Как правило, гораздо более эффективно собирать счета непосредственно в карту и избегать всего ненужного создания Lists.

import scala.collection.TraversableOnce 
import scala.collection.mutable.HashMap 

def counts[T](xs: TraversableOnce[T]): Map[T, Int] = { 
    xs.foldLeft(HashMap.empty[T, Int].withDefaultValue(0))((acc, x) => { acc(x) += 1; acc}).toMap 
} 

После того, как вы определили метод counts вы можете применить его к итератора пар ключ-значение, например, так:

val iter: Iterator[(String, String)] = ??? 
val keyCounts = counts(iter.map(_._1)) 

Метод counts определено выше работает хорошо Iterators над большим числом значения, например

val iter = Iterator.range(0, 100000000).map(i => (i % 1931, i)) 
val countMap = counts(iter.map(_._1)) 
// Map(645 -> 51787, 892 -> 51787, 69 -> 51787, 1322 -> 51786, ...) 

работает отлично, в то время как подход, предложенный в ответ Петра, то есть

val iter = Iterator.range(0, 100000000).map(i => (i % 1931, i)) 
val countMap = iter.toList.groupBy(_._1).mapValues(_.length).toMap 

пыхтит прочь на некоторое время, и в конечном итоге приводит к OutOfMemoryError. Причина, по которой он терпит неудачу, - это из-за ненужного создания List.