pyspark.sql选择其中三列

在PySpark中，如果你想要从一个DataFrame中选择特定的三列，你可以使用`select`方法。`select`方法允许你指定一个或多个列名作为参数，然后返回一个新的DataFrame，其中只包含你指定的列。

下面是一个简单的例子，演示如何使用`select`方法选择三列：

from pyspark.sql import SparkSession

# 创建SparkSession
spark = SparkSession.builder.appName("SelectColumnsExample").getOrCreate()

# 假设有一个DataFrame df，包含列"A", "B", "C", "D"
data = [("a", "b", "c", "d"), ("e", "f", "g", "h")]
columns = ["A", "B", "C", "D"]
df = spark.createDataFrame(data, columns)

# 选择"A", "B", "C"这三列
selected_df = df.select("A", "B", "C")

# 展示选择后的DataFrame
selected_df.show()

在这个例子中，`df.select("A", "B", "C")`会创建一个新的DataFrame，只包含原始DataFrame中的"A", "B", "C"三列。