由于您希望创建一个新行而不考虑列类型，因此可以编写一个函数，该函数将列名作为输入，并返回一个字典，其中包含新行的所有必要信息，包括该列中的项数。
要创建如图所示的输出pyspark Dataframe ，每一列都必须是字符串类型，因为新行中的lastname, gender, salary列必须包含空字符串''。（请参阅here），因此当您在df2和total_row_df之间建立并集时，total_row_df中字串类型的任何数据行都会在产生的数据框中胁迫转换为字串类型。

from pyspark.sql.functions import count

def create_total_row(col_name):
    total_row = {}
    for col in df2.columns:
        if col == 'S':
            total_row[col] = 'T'
        elif col == col_name:
            total_row[col] = df2.select(count(df2[col_name])).collect()[0][0]
        else:
            total_row[col] = ''
    return total_row

total_row = create_total_row('firstname')
total_row_df = spark.createDataFrame([total_row])
df2.union(total_row_df).show()

结果：

+---+---------+--------+------+------+
|  S|firstname|lastname|gender|salary|
+---+---------+--------+------+------+
|  D|    James|   Smith|     M|  3000|
|  D|     Anna|    Rose|     F|  4100|
|  D|   Robert|Williams|     M|  6200|
|  T|        3|        |      |      |
+---+---------+--------+------+------+