Как удалить все данные до n-го вхождения разделителя в строке?

Question

У меня есть строка, разделенная запятой.Я хочу удалить все данные перед моей второй запятой вместе с запятой.Например:

Input: 1,2,3,4,5

Output: 3,4,5

Answer 1

Просто определите UDF, который делает это, что-то вроде этого:

val dropTwoEntries = udf((s: String) => s.split(",").drop(2).mkString(","))
df.withColumn("output", dropTwoEntries(col("input")))