在简单的地图操作

Question

我现在正在使用GraphX进行一些操作，想要这样的东西

val ans = graph.triplets.map(
    e => {
        if (conditon1){
            return ans_1 to RDD_1
        }
        else (condition2){
            return ans_2 to RDD_2
        }
    }
)

我知道我可以使用graph.triplets.map的双重运行来返回2个不同的RDD，就像这样

val RDD_1 = graph.triplets.map(
    e => {
        if (conditon1){
            return ans_1
        }
    })
val RDD_2 = graph.triplets.map(
    e => {
        if (condition2){
            return ans_2
        }
    })

然而，为了提高效率，我想在一次运行中完成它，如上所述。我怎样才能实现它？