pyspark的dataframe如何用[]直接保留几列

您好！对于 PySpark DataFrame，可以使用 select 函数来选择需要保留的列。例如，如果要保留名为 "col1" 和 "col2" 的两列，可以这样操作：

from pyspark.sql.functions import col

df = df.select(col("col1"), col("col2"))

注意，“col1” 和 “col2” 是列名，需要替换为实际的列名。希望这可以帮助回答您的问题！