PySpark当列表

Question

以下项目是我想要实现的操作：

types = ["200","300"] 
def Count(ID): 
    cnd = F.when((**F.col("type") in types**), 1).otherwise(F.lit(0)) 
    return F.sum(cnd).alias("CountTypes") 

粗体语法是不正确的，任何建议，如何在这里得到正确的语法PySpark？

Answer 1

我不知道你正在努力实现什么，但这里是正确的语法：

>>> types = ["200","300"] 
>>> from pyspark.sql import functions as F 
>>> cnd = F.when(F.col("type").isin(types),1).otherwise(F.lit(0)) 
>>> F.sum(cnd).alias("count_types") 
# Column<(sum(CASE WHEN type IN (200,300) THEN 1 ELSE 0),mode=Complete,isDistinct=false) AS count_types#3>