создать столбец с длиной строки в другом столбце pyspark

Question

У меня есть столбец в фрейме данных в pyspark, например, «Col1» ниже. Я хотел бы создать новый столбец «Col2» с длиной каждой строки из «Col1». Я новичок в pyspark, я гуглял, но не видел примеров того, как это сделать. Любые советы очень ценятся.

пример:

Col1 Col2
12   2
123  3

Answer 1

Вы можете использовать функцию length:

import pyspark.sql.functions as F
df.withColumn('Col2', F.length('Col1')).show()
+----+----+
|Col1|Col2|
+----+----+
|  12|   2|
| 123|   3|
+----+----+