如何根据另一列的值创建一列？

Question

我有一个称为code的列，此列包含2个字母值。 我想做的是创建另一个名为with_j列，在该列中，我希望所有以j结尾的代码。 但是，与其说实际的代码，不如说： *j

所以看起来像这样：

之前：

code
bx
aj
dj
la

后

code   with_j
bx
aj     *j
dj     *j
la     

有办法吗？

Answer 1

使用endsWith和when/otherwise实际上非常简单。

Seq("bx", "aj", "dj", "la")
    .toDF("code")
    .withColumn("with_j", 
        when('code endsWith "j", "*j").otherwise("")
    )
    .show