要按when表达式分组并求和计数：

from pyspark.sql import functions as F

df1 = df.groupBy(
    when(
        F.col("type").isin("A", "B"), F.col("type")
    ).otherwise("Others").alias("type")
).agg(
    F.sum("count").alias("count")
)

df1.show()

# +------+-----+

# |  type|count|

# +------+-----+

# |     B| 5000|

# |     A| 5000|

# |Others|  776|

# +------+-----+

您可以根据类型将Dataframe分为两部分，为第二部分聚合一个和，然后执行 unionAll 把它们结合起来。

import pyspark.sql.functions as F

result = df.filter("type in ('A', 'B')").unionAll(
    df.filter("type not in ('A', 'B')")
      .select(F.lit('Other'), F.sum('count'))
)

result.show()
+-----+-----+
| type|count|
+-----+-----+
|    A| 5000|
|    B| 5000|
|Other|  776|
+-----+-----+