我的数据类型是:List[(Option[Set[Long]], Option[Set[(Long, Long)]])]
(即一个选项的2元组列表,一个用于一组Long,一个为一组Longs)。例如:
val l = List((Option(Set(1L, 2L, 3L)), Option(Set((4L, 5L), (6L, 7L)))))
我希望将此List平面映射为List[Set[Long]]
类型,使第二个Option的每个第一个元素与第一个Option连接到一个新的Set。所以这里想要的结果是:
List(Set(1L, 2L, 3L, 4L), Set(1L, 2L, 3L, 6L))
这适用于以下代码:
l.flatMap {
case(setOpt1, setOpt2) =>
val set1 = setOpt1.getOrElse(Set[Long]());
val set2 = setOpt2.getOrElse(Set[(Long, Long)]());
set2.map(k => set1 ++ Set(k._1))
}
当第二个选项为None
,并且第一个选项"存在"时,图变厚,例如:
val l = List((Option(Set(1L, 2L, 3L)), None))
假设我没有办法避免这种情况。在这种情况下,我想得到第一个设置如下:
List(Set(1L, 2L, 3L))
然而,使用与上面相同的代码,set2
值变成了一个空集,在这种情况下,map
没有任何意义,我会得到一个空列表:
res60: List[scala.collection.immutable.Set[Long]] = List()
那么,我该怎么做呢?最好优雅:)
您可以添加case语句来处理Option
的每个组合
def g(l:List[(Option[Set[Long]], Option[Set[(Long, Long)]])]) = l.flatMap{
case (Some(single), None) =>
List(single)
case (Some(single), Some(tuple)) =>
tuple.map(k => single ++ Set(k._1))
case (None, Some(tuple)) =>
tuple.map(k => Set(k._1))
case (None, None) =>
List()
}
这将产生您所期望的结果。以下是我运行的所有测试:
val l1 = List((Option(Set(1L, 2L, 3L)), Option(Set((4L, 5L), (6L, 7L)))))
g(l1)
res0: List[Set[Long]] = List(Set(1, 2, 3, 4), Set(1, 2, 3, 6))
val l2 = List((Option(Set(1L, 2L, 3L)), None))
g(l2)
res1: List[Set[Long]] = List(Set(1, 2, 3))
val l3 = List((None, Option(Set((4L, 5L), (6L, 7L)))))
g(l3)
res2: List[Set[Long]] = List(Set(4), Set(6))
val l4 = List((None, None))
g(l4)
res3: List[Set[Long]] = List()