根据列表值更新列Dataframe列

Question

我有一个spark数据帧列id和articles，还有3个列表，如下所示。a_list，b_list，c_list。

a_list=[4, 10], b_list=[11,3], c_list=[3,6]

df = spark.createDataFrame([(1, 4), (2, 3), (5, 6)], ("id", "articles"))

我想根据数据框列Found的值与列表articles之间的匹配来更新列(a_list, b_list,c_list)

当前，我只能在有2个列表的情况下执行以下代码。

import pyspark.sql.functions as func                    
df.withColumn('E', func.when(df.articles.isin(a_list), 'Found in a_list').otherwise('Found in b_list'))            

如何推断两个以上的列表？

预期输出

+---+--------+---------------+                                                  
| id|articles|          Found|
+---+--------+---------------+
|  1|       4|Found in a_list|
|  2|       3|Found in b_list|
|  5|       6|Found in c_list|
+---+--------+---------------+