Pyspark - Sql 过滤器 - 通过检查 id 值是否出现在数组中来选择所有行

Question

我在 Pyspark 中有一个 DataFrame，我需要选择其中 id 值出现在数组中的行。有人可以帮我吗？

例子：

 +---+-----+
 | id| col2|
 +---+-----+
 |123|  2  |
 |245| 32  |
 | 12| 34  |
 |234|  1  |
 +---+-----+

数组：[123, 12, 234]

想要的结果：

+---+-----+
| id| col2|
+---+-----+
|123|  2  |
| 12| 34  |
|234|  1  |
+---+-----+

Answer 1

你可以用isin与filter：

ids = [123, 12, 234]
df.filter(df.id.isin(ids)).show()
+---+----+
| id|col2|
+---+----+
|123|   2|
| 12|  34|
|234|   1|
+---+----+