How to convert a Seq[A] to a Map[Int, A] using a value of A as the key in the map?
I have a Seq
containing objects of a class that looks like this:
class A (val key: Int, ...)
Now I want to convert this Seq
to a Map
, using the key
value of each object as the key, and the object itself as the value. So:
val seq: Seq[A] = ...
val map: Map[Int, A] = ... // How to convert seq to map?
How can I does this efficiently and in an elegant way in Scala 2.8?
Since 2.8 Scala has had .toMap
, so:
val map = seq.map(a => a.key -> a).toMap
or if you're gung ho about avoiding constructing an intermediate sequence of tuples, then in Scala 2.8 through 2.12:
val map: Map[Int, A] = seq.map(a => a.key -> a)(collection.breakOut)
or in Scala 2.13 and 3 (which don't have breakOut
, but do have a reliable .view
):
val map = seq.view.map(a => a.key -> a).toMap
Map over your Seq
and produce a sequence of tuples. Then use those tuples to create a Map
. Works in all versions of Scala.
val map = Map(seq map { a => a.key -> a }: _*)
One more 2.8 variation, for good measure, also efficient:
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)))
Note that if you have duplicate keys, you'll discard some of them during Map creation! You could use groupBy
to create a map where each value is a sequence:
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))))
As scala knows to convert a Tuple of two to a map, you would first want to convert your seq to a tuple and then to map so (doesn't matter if it's int, in our case string, string):
The general algorithm is this:
- For each item in Seq
- Item --> Tuple(key, value)
- For each tuple(key, value)
- Aggregate to Map(key,value)
Or to sum up:
Step 1: Seq --> Tuple of two
Step 2: Tuple of two --> Map
Example:
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)