Как преобразовать Seq [A] в карту [Int, A], используя значение A в качестве ключа на карте?

У меня есть объект Seq, содержащий объект класса, который выглядит следующим образом:

class A (val key: Int, ...)

Теперь я хочу преобразовать этот Seq в Map, используя значение key каждого объекта в качестве ключа, а сам объект - как значение. Итак:

val seq: Seq[A] = ...
val map: Map[Int, A] = ... // How to convert seq to map?

Как я могу сделать это эффективно и элегантно в Scala 2.8?

Ответы

Ответ 1

Наведите курсор на Seq и создайте последовательность кортежей. Затем используйте эти кортежи для создания Map. Работает во всех версиях Scala.

val map = Map(seq map { a => a.key -> a }: _*)

Ответ 2

С .toMap 2.8 у Scala был .toMap, поэтому:

val map = seq.map(a => a.key -> a).toMap

или если вы не хотите создавать промежуточную последовательность кортежей:

val map: Map[Int, A] = seq.map(a => a.key -> a)(collection.breakOut)

Ответ 3

Еще одна вариация 2.8 для хорошей меры также эффективна:

scala> case class A(key: Int, x: Int)
defined class A

scala> val l = List(A(1, 2), A(1, 3), A(2, 1))
l: List[A] = List(A(1,2), A(1,3), A(2,1))

scala> val m: Map[Int, A] = (l, l).zipped.map(_.key -> _)(collection.breakOut)
m: Map[Int,A] = Map((1,A(1,3)), (2,A(2,1)))

Обратите внимание, что если у вас есть повторяющиеся ключи, вы отбросите некоторые из них во время создания Карты! Вы можете использовать groupBy для создания карты, где каждое значение является последовательностью:

scala> l.groupBy(_.key)
res1: scala.collection.Map[Int,List[A]] = Map((1,List(A(1,2), A(1,3))), (2,List(A(2,1))))

Ответ 4

Поскольку Scala знает, как преобразовать кортеж из двух в карту, вы сначала захотите преобразовать свой seq в кортеж, а затем отобразить его (не имеет значения, если это int, в нашем случае string, string):

Общий алгоритм таков:

  1. Для каждого элемента в Seq
  2. Item → Tuple (ключ, значение)
  3. Для каждого кортежа (ключ, значение)
  4. Агрегировать на карту (ключ, значение)

Или подвести итог:

Шаг 1: Seq → кортеж из двух

Шаг 2: Кортеж из двух → Карта

Пример:

case class MyData(key: String, value: String) // One item in seq to be converted to a map entry.

// Our sequence, simply a seq of MyData
val myDataSeq = Seq(MyData("key1", "value1"), MyData("key2", "value2"), MyData("key3", "value3")) // List((key1,value1), (key2,value2), (key3,value3))

// Step 1: Convert seq to tuple
val myDataSeqAsTuple = myDataSeq.map(myData => (myData.key, myData.value)) // List((key1,value1), (key2,value2), (key3,value3))

// Step 2: Convert tuple of two to map.
val myDataFromTupleToMap = myDataSeqAsTuple.toMap // Map(key1 -> value1, key2 -> value2, key3 -> value3)