選取

投影一組表達式並回傳新的 DataFrame。

語法

select(*cols: "ColumnOrName")

參數

參數 類型 說明
cols str、欄位或列表 欄位名稱(字串)或表達式(欄位)。 若其中一欄名稱為 '*',該欄會展開以包含目前 DataFrame 中的所有欄位。

退貨

DataFrame:一個包含子集(或全部)欄位的資料框架。

Examples

df = spark.createDataFrame([
    (2, "Alice"), (5, "Bob")], schema=["age", "name"])

df.select('*').show()
# +---+-----+
# |age| name|
# +---+-----+
# |  2|Alice|
# |  5|  Bob|
# +---+-----+

df.select(df.name, (df.age + 10).alias('age')).show()
# +-----+---+
# | name|age|
# +-----+---+
# |Alice| 12|
# |  Bob| 15|
# +-----+---+