Как сделать столбец на основе значения другого столбца?

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