查找索引*哪里* [英] Find Indexes *Where*

问题描述

Vector 中有一个 indexWhere 函数，用于查找匹配项的索引.

There's a indexWhere function in Vector that finds the index of a match.

def indexWhere(p: (A) ⇒ Boolean, from: Int): Int
> Finds index of the first element satisfying some predicate after or 
> at some start index.

http://www.scala-lang.org/api/current/index.html#scala.collection.immutable.Vector

我编写了这个函数来查找发生这种匹配的所有索引.

I wrote this function to find all indexes where such a match occurs.

  def getAllIndexesWhere[A,B](as: List[A])(f: (B => Boolean))(g: A => B): Vector[B] = {
    def go(y: List[A], acc: List[Option[B]]): Vector[B] = as match {
      case x :: xs => val result = if (f(g(x))) Some(g(x)) else None
                      go(xs, acc :+ result)
      case Nil => acc.flatten.toVector
    }
    go(as, Nil)
  }

但是，是否已经有集合的内置函数?

However, is there already a built-in function of a collection?

推荐答案

zipWithIndex、filter 和 map 是可以组合起来得到某个谓词的所有索引.

zipWithIndex, filter, and map are built-ins that can be combined to get all the indices of some predicate.

获取列表中偶数值的索引.

Get the indices of the even values in the list.

scala> List(1,2,3,4,5,6,7,8,9,10).zipWithIndex.filter(_._1 % 2 == 0).map(_._2)
res0: List[Int] = List(1, 3, 5, 7, 9)

您也可以使用 collect 作为 @0__ 笔记.

You can also use collect as @0__ notes.

scala> List(1,2,3,4,5,6,7,8,9,10).zipWithIndex.collect{ case(a,b) if a % 2 == 0 => b}
res1: List[Int] = List(1, 3, 5, 7, 9)

这篇关于查找索引*哪里*的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！