我需要在Scala中创建一个新的DF并附加我已经拥有的行。我该怎么做呢?

时间:2019-03-16 02:29:13

标签: scala apache-spark

我需要使用带有字符串“ v”,“ d”,“ c”的列标题创建df

我有3行:

in_max =[a,b,c]
out_max = [d,e,f]
weight_max = [g,h]

我想设置我的df,使其看起来像这样:

"v","d", "c"

a,b,"i"

d,f,"o"

g,h, "t"

基本上第一行是in_max中的前两个值,后跟一个“ i” 第二行是out_max中的第一个和第三个值,后跟一个“ o” 最后两个值是weight_max中的值,后跟一个“ t”。

我该如何在Scala中做到这一点?

1 个答案:

答案 0 :(得分:1)

让我们简单地将您的逻辑转换为代码:

val in_max = Seq("a","b","c")
val out_max = Seq("d","e","f")
val weight_max = Seq("g","h")

val df = Seq((in_max(0), in_max(1), "i"),
             (out_max(0), out_max(2), "o"),
             (weight_max(0), weight_max(1), "t"))
     .toDF("v", "d", "c")

df.show
+---+---+---+
|  v|  d|  c|
+---+---+---+
|  a|  b|  i|
|  d|  f|  o|
|  g|  h|  t|
+---+---+---+