scala - 从 2 个给定列表中获取公共元素

Question

所以我有这个 2 collections：

import collection.mutable.ListBuffer

val list1 = ListBuffer[(String, String)]()
list1 += (("Italy", "valid"))
list1 += (("Germany", "not valid"))
list1 += (("USA", "not valid"))
list1 += (("Romania", "valid"))

val list2 = ListBuffer[String]()
list2 += "Germany"
list2 += "USA"
list2 += "Romania"
list2 += "Italy"
list2 += "France"
list2 += "Croatia"

我想获得包含特定国家的新列表，condition例如valid，结果将是新列表：

Italy, Romania

Answer 1

当您需要时filter，map请同时使用collect：

list1.collect { case (c, "valid") => c } intersect list2

Answer 2

使用 for 理解如下，

for ((a,"valid") <- list1 if list2.contains(a)) yield a

这种理解去糖化为flatMap一个惰性过滤器。我们模式匹配（提取）第二个项目匹配的那些元组，valid"并检查验证的国家是否属于第二个列表。

Answer 3

我认为您应该过滤，映射然后相交

list1 filter(_._2=="valid") map(_._1) intersect list2

Answer 4

您可以定义以下方法：

def intersectWithCond[A, B](xs: Seq[(A, B)], ys: Seq[A])(cond: B => Boolean): Seq[A] = {
  xs.filter(x => cond(x._2)).map(_._1).intersect(ys)
}

用法：

intersectWithCond(list1, list2)(_ == "valid") // returns ListBuffer(Italy, Romania)