selecionar

Projeta um conjunto de expressões e retorna um novo DataFrame.

Sintaxe

select(*cols: "ColumnOrName")

Parâmetros

Parâmetro Tipo Descrição
cols str, Column ou list nomes de colunas (string) ou expressões (Column). Se um dos nomes das colunas for '*', essa coluna é expandida para incluir todas as colunas do DataFrame atual.

Devoluções

DataFrame: Um DataFrame com subconjunto (ou todos) de colunas.

Exemplos

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|
# +-----+---+