Commit 5455edea authored by 王志伟's avatar 王志伟

取icon训练数据

parent 54b12b08
...@@ -540,13 +540,11 @@ object icon_train_data { ...@@ -540,13 +540,11 @@ object icon_train_data {
test_diary.show() test_diary.show()
val Df2 = test_diary.withColumn("temp",split("tag_list", ",")) // val Df2 = test_diary.withColumn("tag_list",explode(split()))
.select( // Df2.show()
test_diary("temp").getItem(0).as("col1"), import sc.implicits._
test_diary("temp").getItem(1).as("col2"), val test=test_diary.rdd.map(row=>(row(0).toString,row(1).toString))
test_diary("temp").getItem(2).as("col3") .map(row=>(row._1,row._2.split(",").head,row._2.split(",")(1),row._2.split(",")(2))).toDF().show()
).drop("temp")
Df2.show()
// val df_explode = test_diary.withColumn("e", explode(split(test_diary['tag_list'], ","))).drop("tag_list") // val df_explode = test_diary.withColumn("e", explode(split(test_diary['tag_list'], ","))).drop("tag_list")
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment