使用pyspark计算文本文件中所有单词的长度总和问题

from pyspark.sql import SparkSession
from pyspark.sql.functions import length, sum

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

text_df = spark.read.text("path/to/textfile.txt")

word_length_df = text_df.select(length(text_df.value).alias("word_length"))

total_length = word_length_df.select(sum(word_length_df.word_length)).collect()[0][0]

print("所有单词的长度总和为:", total_length)