如何在pyspark中更新dataframe中结构数组中的值？

from pyspark.sql.functions import col, expr, when

# 假设存在一个名为df的DataFrame，包含一个名为struct_array的结构数组列

# 定义要更新的元素的索引和新值
update_index = 1
new_value = "new_value"

# 使用withColumn方法更新结构数组的值
df_updated = df.withColumn("struct_array", expr("transform(struct_array, (x, i) -> " +
                                               "when(i = {0}, struct(x.field1, '{1}')).otherwise(x)))"
                                               .format(update_index, new_value))

df_updated.show()