Spark Dataframe column with last character of other column

Question

I\'m looking for a way to get the last character from a string in a dataframe column and place it into another column.

I have a Spark dataframe that looks like this:

Answer 1

Just use the substring function

from pyspark.sql.functions import substring
df.withColumn("b", substring(col("columnName"), -1, 1))

Answer 2

One way is by using Column substr() function:

df = df.withColumn("lastchar", df.animal.substr(-1,1))

See documentation: https://spark.apache.org/docs/2.1.0/api/python/pyspark.sql.html#pyspark.sql.Column.substr