如何计算PySpark中嵌套列表中重复元素的个数？

from pyspark.sql import SparkSession
from pyspark.sql.functions import explode, col, count

spark = SparkSession.builder.appName("NestedListCount").getOrCreate()

data = [("A", [[1, 2, 3], [2, 3, 4], [1, 2, 3]]),
        ("B", [[4, 5, 6], [5, 6, 7], [4, 5, 6]])]
df = spark.createDataFrame(data, ["id", "nested_list"])

df_exploded = df.select("id", explode("nested_list").alias("element"))

result = df_exploded.groupBy("id", "element").agg(count("*").alias("count"))

result.show()