apache spark - How to split a number and add hyphen in a pyspark dataframe?

Question

I want to split all numbers in a column.

number: 123456789012

to be 123-4567890-12

So I want to add - at positions 4 and 10.

I do not want to make new columns

question from:https://stackoverflow.com/questions/66066429/how-to-split-a-number-and-add-hyphen-in-a-pyspark-dataframe

Answer 1

You can use regexp_replace :

from pyspark.sql import functions as F

df1 = df.withColumn(
    "number",
    F.regexp_replace(F.col("number"), "(\d{3})(\d{7})(\d+)", "$1-$2-$3")
)

df1.show()

#+--------------+
#|        number|
#+--------------+
#|123-4567890-12|
#+--------------+