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

Question

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

Answer 1

Я полагаю, вы ищете 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) )

Answer 2

Использование. 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)")

Answer 3

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

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. Наслаждайтесь!

Answer 4

В 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)

Answer 5

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

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

Answer 6

Использование .map в .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
     )

Answer 7

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

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