Question

我有SortedMap类型：

data: SortedMap[Long, SortedMap[String, Double]]

如何找到特定密钥的索引。

例如：

data = (1L -> ("a" -> 1.), 2L -> ("b" -> 1., "c" -> 2.), 3L -> ("b" -> 1.))

我想找到键2L的索引（结果应为1）。

Answer 1

您可以使用zipWithIndex将每个键值与其索引配对，然后使用collectFirst找到具有正确键的对并提取索引。这将返回Option[Int]（如果找不到密钥，则为None。）

val data: SortedMap[Long, SortedMap[String, Double]] = SortedMap(
    1L -> SortedMap("a" -> 1.0), 
    2L -> SortedMap("b" -> 1.0, "c" -> 2.0), 
    3L -> SortedMap("b" -> 1.0)
)

def findIndex(data: SortedMap[Long, SortedMap[String, Double]], key: Long): Option[Int] = 
    data.zipWithIndex.collectFirst { case ((`key` , _), i) => i }

scala> findIndex(data, 1L)
res16: Option[Int] = Some(0)

scala> findIndex(data, 2L)
res17: Option[Int] = Some(1)

scala> findIndex(data, 10L)
res18: Option[Int] = None

Answer 2

基于密钥的迭代器发现了这种方法，可以避免索引中间集合，如下所示，

data.keysIterator.indexWhere(_ == 2)

为了便于使用，请考虑这个隐含的

implicit class RichSortedMap[A,B](val m: SortedMap) extends AnyVal {
  def keyIndexWhere(k: A) = m.keysIterator.indexWhere( _ == k )
}

所以你可以按如下方式使用它，

data.keyIndexWhere(2)

查找特定SortedMap的索引

2 个答案: