from pyspark.sql.functions import col, rowNumber
from pyspark.sql.window import Window
w = Window.orderBy()
indexed = df.withColumn('index', rowNumber().over(w))
indexed.where(col('index').isin(set(indexes)))
indexed.where('index in ({0})'.format(','.join(str(x) for x in indexes)))
PySpark DataFrames - способ перечисления без преобразования в Pandas?