Scala лучший способ превратить коллекцию в Map-by-key?

Если у меня есть коллекция c типа T, и есть свойство p на T (типа p, скажем), что лучший способ сделать карту по байту, извлечения ключа?

val c: Collection[T]
val m: Map[P, T]

Один из способов:

m = new HashMap[P, T]
c foreach { t => m add (t.getP, t) }

Но теперь мне нужна изменчивая карта. Есть ли лучший способ сделать это, чтобы он в 1 строке, и я получаю неизменную карту? (Очевидно, я мог бы превратить это в простую библиотечную утилиту, как и в Java, но я подозреваю, что в Scala нет необходимости)

Ответ 1

Вы можете использовать

c map (t => t.getP -> t) toMap

но имейте в виду, что для этого требуется 2 обхода.

Ответ 2

Вы можете построить карту с переменным количеством кортежей. Поэтому используйте метод map в коллекции, чтобы преобразовать его в набор кортежей, а затем использовать трюк: _ *, чтобы преобразовать результат в переменный аргумент.

scala> val list = List("this", "maps", "string", "to", "length") map {s => (s, s.length)}
list: List[(java.lang.String, Int)] = List((this,4), (maps,4), (string,6), (to,2), (length,6))

scala> val list = List("this", "is", "a", "bunch", "of", "strings")
list: List[java.lang.String] = List(this, is, a, bunch, of, strings)

scala> val string2Length = Map(list map {s => (s, s.length)} : _*)
string2Length: scala.collection.immutable.Map[java.lang.String,Int] = Map(strings -> 7, of -> 2, bunch -> 5, a -> 1, is -> 2, this -> 4)

Ответ 3

В дополнение к решению @James Iry, это также возможно сделать, используя складку. Я подозреваю, что это решение немного быстрее, чем метод кортежа (меньше объектов мусора создаются):

val list = List("this", "maps", "string", "to", "length")
val map = list.foldLeft(Map[String, Int]()) { (m, s) => m(s) = s.length }

Ответ 4

Это может быть реализовано неизменно и с одним обходным путем путем сгибания коллекции следующим образом.

val map = c.foldLeft(Map[P, T]()) { (m, t) => m + (t.getP -> t) }

Решение работает, потому что добавление к неизменяемой карте возвращает новую неизменяемую карту с дополнительной записью, и это значение служит в качестве аккумулятора через операцию сгиба.

Компромисс заключается в простоте кода и его эффективности. Таким образом, для больших коллекций этот подход может быть более подходящим, чем использование двух реализаций прохождения, таких как применение map и toMap.

Ответ 5

Другое решение (может не работать для всех типов)

import scala.collection.breakOut
val m:Map[P, T] = c.map(t => (t.getP, t))(breakOut)

это позволяет избежать создания списка посредников, подробнее здесь: Scala 2.8 breakOut

Ответ 6

То, что вы пытаетесь достичь, немного неопределенно.
Что если два или более элементов в c имеют одинаковый p? Какой элемент будет сопоставлен с этим p на карте?

Более точный способ взглянуть на это - получить карту между p и всеми c объектами, у которых она есть:

val m: Map[P, Collection[T]]

Это может быть легко достигнуто с K):scala.collection.GenMap[K,Repr] rel=noreferrer>groupBy:

val m: Map[P, Collection[T]] = c.groupBy(t => t.p)

Если вам все еще нужна исходная карта, вы можете, например, сопоставить p первому t, в котором она есть:

val m: Map[P, T] = c.groupBy(t => t.p) map { case (p, ts) =>  p -> ts.head }

Ответ 7

Это, вероятно, не самый эффективный способ превратить список в карту, но он делает код вызова более читабельным. Я использовал неявные преобразования, чтобы добавить метод mapBy в список:

implicit def list2ListWithMapBy[T](list: List[T]): ListWithMapBy[T] = {
  new ListWithMapBy(list)
}

class ListWithMapBy[V](list: List[V]){
  def mapBy[K](keyFunc: V => K) = {
    list.map(a => keyFunc(a) -> a).toMap
  }
}

Пример кода вызова:

val list = List("A", "AA", "AAA")
list.mapBy(_.length)                  //Map(1 -> A, 2 -> AA, 3 -> AAA)

Обратите внимание, что из-за неявного преобразования код вызывающего абонента должен импортировать scala implicitConversions.

Ответ 8

c map (_.getP) zip c

Хорошо работает и очень интуитивно

Ответ 9

Для чего это стоит, вот два бессмысленных способа сделать это:

scala> case class Foo(bar: Int)
defined class Foo

scala> import scalaz._, Scalaz._
import scalaz._
import Scalaz._

scala> val c = Vector(Foo(9), Foo(11))
c: scala.collection.immutable.Vector[Foo] = Vector(Foo(9), Foo(11))

scala> c.map(((_: Foo).bar) &&& identity).toMap
res30: scala.collection.immutable.Map[Int,Foo] = Map(9 -> Foo(9), 11 -> Foo(11))

scala> c.map(((_: Foo).bar) >>= (Pair.apply[Int, Foo] _).curried).toMap
res31: scala.collection.immutable.Map[Int,Foo] = Map(9 -> Foo(9), 11 -> Foo(11))

Ответ 10

Просто сделай это.

c.map.(x => x.getP -> x).toMap

Ответ 11

Как насчет использования zip и toMap?

myList.zip(myList.map(_.length)).toMap

Ответ 12

Это работает для меня:

val personsMap = persons.foldLeft(scala.collection.mutable.Map[Int, PersonDTO]()) {
    (m, p) => m(p.id) = p; m
}

Карта должна быть изменчивой, и Карта должна быть возвращена, поскольку добавление к изменяемой карте не возвращает карту.

Ответ 13

использовать map() для коллекции с последующей toMap

val map = list.map(e => (e, e.length)).toMap