如何将常量整数值添加到整数列？

Question

我有一个包含3列的数据框，其中一列是整数。我想为它添加一个整数值。

id person age
1  person1 4
2  person4 5
3  person3 7

我想在他们的年龄增加一个20岁，以便他们的年龄分别为24,25和27岁。

Answer 1

您可以添加新列，并按年龄

在年龄列中添加20

dataFrame.withColumn("age", col("age") + 20)

希望这有帮助！

Answer 2

您可以使用UserDefinedFunction来计算可能更复杂的表达式。在Java 8中，它看起来可能像这样：

    UserDefinedFunction doAge = udf(
            (Integer age) -> age+20, DataTypes.IntegerType
    );

    dataFrame = dataFrame.withColumn("age",doAge.apply(dataFrame.col("age")));