Я предполагаю, что в вашем фрейме данных есть столбец "body", представляющий собой строку json, и вы хотите проанализировать json и извлечь из него элемент.
Сначала необходимо определить или извлечьСхема JSON.А затем проанализируйте строку json и извлеките ее элементы в виде столбца.Из извлеченных столбцов вы можете выбрать нужные столбцы.
json_schema = spark.read.json(df.rdd.map(lambda row: row.body)).schema
df2 = df.withColumn('body_json', F.from_json(F.col('body'), json_schema))\
.select("body_json.*").select('CustomerType')
display(df2)