[英]Spark, add new Column with the same value in Scala
I have some problem with the withColumn
function in Spark-Scala environment. 我在Spark-Scala环境中的
withColumn
函数有一些问题。 I would like to add a new Column in my DataFrame like that: 我想在我的DataFrame中添加一个新列,如下所示:
+---+----+---+
| A| B| C|
+---+----+---+
| 4|blah| 2|
| 2| | 3|
| 56| foo| 3|
|100|null| 5|
+---+----+---+
became: 成为:
+---+----+---+-----+
| A| B| C| D |
+---+----+---+-----+
| 4|blah| 2| 750|
| 2| | 3| 750|
| 56| foo| 3| 750|
|100|null| 5| 750|
+---+----+---+-----+
the column D in one value repeated N-time for each row in my DataFrame. 对于我的DataFrame中的每一行,一列中的列D重复N次。
The code are this: 代码是这样的:
var totVehicles : Double = df_totVehicles(0).getDouble(0); //return 750
The variable totVehicles returns the correct value, it's works! 变量totVehicles返回正确的值,它的工作原理!
The second DataFrame has to calculate 2 fields (id_zipcode, n_vehicles), and add the third column (with the same value -750): 第二个DataFrame必须计算2个字段(id_zipcode,n_vehicles),并添加第三列(具有相同的值-750):
var df_nVehicles =
df_carPark.filter(
substring($"id_time",1,4) < 2013
).groupBy(
$"id_zipcode"
).agg(
sum($"n_vehicles") as 'n_vehicles
).select(
$"id_zipcode" as 'id_zipcode,
'n_vehicles
).orderBy(
'id_zipcode,
'n_vehicles
);
Finally, I add the new column with withColumn
function: 最后,我使用
withColumn
函数添加新列:
var df_nVehicles2 = df_nVehicles.withColumn(totVehicles, df_nVehicles("n_vehicles") + df_nVehicles("id_zipcode"))
But Spark returns me this error: 但Spark告诉我这个错误:
error: value withColumn is not a member of Unit
var df_nVehicles2 = df_nVehicles.withColumn(totVehicles, df_nVehicles("n_vehicles") + df_nVehicles("id_zipcode"))
Can you help me? 你能帮助我吗? Thank you very much!
非常感谢你!
lit
function is for adding literal values as a column lit
函数用于将文字值添加为列
import org.apache.spark.sql.functions._
df.withColumn("D", lit(750))
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.