it-roy-ru.com

Как я могу использовать карту и получить индекс в Scala?

Есть ли встроенный List/Sequence, который ведет себя как map и также предоставляет индекс элемента?

89
Geo

Я полагаю, вы ищете zipWithIndex?

scala> val ls = List("Mary", "had", "a", "little", "lamb")
scala> ls.zipWithIndex.foreach{ case (e, i) => println(i+" "+e) }
0 Mary
1 had
2 a
3 little
4 lamb

От: http://www.artima.com/forums/flat.jsp?forum=283&thread=243570

У вас также есть варианты, такие как: 

for((e,i) <- List("Mary", "had", "a", "little", "lamb").zipWithIndex) println(i+" "+e)

или же:

List("Mary", "had", "a", "little", "lamb").zipWithIndex.foreach( (t) => println(t._2+" "+t._1) )
131
Viktor Klang

Используйте . map in . zipWithIndex

val myList = List("a", "b", "c")

myList.zipWithIndex.map { case (element, index) => 
   println(element, index) 
   s"${element}(${index})"
}

Результат:

List("a(0)", "b(1)", "c(2)")
45
Paulo Cheque

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

myIterable map (doIndexed(someFunction))

Функция doIndexed- оборачивает внутреннюю функцию, которая получает как индекс, так и элементы myIterable. Это может быть знакомо вам по JavaScript.

Вот способ достижения этой цели. Рассмотрим следующую утилиту:

object TraversableUtil {
    class IndexMemoizingFunction[A, B](f: (Int, A) => B) extends Function1[A, B] {
        private var index = 0
        override def apply(a: A): B = {
            val ret = f(index, a)
            index += 1
            ret
        }
    }

    def doIndexed[A, B](f: (Int, A) => B): A => B = {
        new IndexMemoizingFunction(f)
    }
}

Это уже все, что вам нужно. Вы можете применить это, например, следующим образом:

import TraversableUtil._
List('a','b','c').map(doIndexed((i, char) => char + i))

что приводит к списку

List(97, 99, 101)

Таким образом, вы можете использовать обычные Traversable-функции за счет обертывания вашей эффективной функции. Накладные расходы - создание запоминающего объекта и счетчика в нем. В противном случае это решение так же хорошо (или плохо) с точки зрения памяти или производительности, как использование неиндексированного map. Наслаждайтесь!

8
Matt Brasen

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

val ci = List("These","are","words").elements.counted
scala> ci map (i => i+"=#"+ci.count) toList
res0: List[Java.lang.String] = List(These=#0,are=#1,words=#2)
5
Rex Kerr

Или, предполагая, что ваша коллекция имеет постоянное время доступа, вы можете отобразить список индексов вместо фактической коллекции:

val ls = List("a","b","c")
0.until(ls.length).map( i => doStuffWithElem(i,ls(i)) )
3
Cristian Vrabie

Используйте .map in .zipWithIndex со структурой данных карты

val sampleMap = Map("a" -> "hello", "b" -> "world", "c" -> "again")

val result = sampleMap.zipWithIndex.map { case ((key, value), index) => 
    s"Key: $key - Value: $value with Index: $index"
}

Результаты

 List(
       Key: a - Value: hello with Index: 0, 
       Key: b - Value: world with Index: 1, 
       Key: c - Value: again with Index: 2
     )
0
fgfernandez0321

Если вам также требуется поиск по значениям карты (как я должен был):

val ls = List("a","b","c")
val ls_index_map = ls.zipWithIndex.toMap 
0
Yair Beer