sort_array

Ordena o array de entrada por ordem ascendente ou descendente de acordo com a ordenação natural dos elementos do array. Os elementos nulos serão colocados no início do array devolvido por ordem crescente ou no final do array devolvido por ordem decrescente.

Sintaxe

from pyspark.sql import functions as sf

sf.sort_array(col, asc=True)

Parâmetros

Parâmetro	Tipo	Description
col	pyspark.sql.Column ou str	Nome da coluna ou expressão.
asc	bool, opcional	Se deve ordenar por ordem ascendente ou descendente. Se asc for Verdadeiro (por defeito), então a ordenação é por ordem crescente. Se Falso, então por ordem decrescente.

Devoluções

pyspark.sql.Column: Arranjo ordenado.

Examples

Exemplo 1: Ordenar um array por ordem crescente

import pyspark.sql.functions as sf
df = spark.createDataFrame([([2, 1, None, 3],)], ['data'])
df.select(sf.sort_array(df.data)).show()

+----------------------+
|sort_array(data, true)|
+----------------------+
|       [NULL, 1, 2, 3]|
+----------------------+

Exemplo 2: Ordenar um array por ordem decrescente

import pyspark.sql.functions as sf
df = spark.createDataFrame([([2, 1, None, 3],)], ['data'])
df.select(sf.sort_array(df.data, asc=False)).show()

+-----------------------+
|sort_array(data, false)|
+-----------------------+
|        [3, 2, 1, NULL]|
+-----------------------+

Exemplo 3: Ordenar um array com um único elemento

import pyspark.sql.functions as sf
df = spark.createDataFrame([([1],)], ['data'])
df.select(sf.sort_array(df.data)).show()

+----------------------+
|sort_array(data, true)|
+----------------------+
|                   [1]|
+----------------------+

Exemplo 4: Ordenar um array vazio

from pyspark.sql import functions as sf
from pyspark.sql.types import ArrayType, StringType, StructField, StructType
schema = StructType([StructField("data", ArrayType(StringType()), True)])
df = spark.createDataFrame([([],)], schema=schema)
df.select(sf.sort_array(df.data)).show()

+----------------------+
|sort_array(data, true)|
+----------------------+
|                    []|
+----------------------+

Exemplo 5: Ordenar um array com valores nulos

from pyspark.sql import functions as sf
from pyspark.sql.types import ArrayType, IntegerType, StructType, StructField
schema = StructType([StructField("data", ArrayType(IntegerType()), True)])
df = spark.createDataFrame([([None, None, None],)], schema=schema)
df.select(sf.sort_array(df.data)).show()

+----------------------+
|sort_array(data, true)|
+----------------------+
|    [NULL, NULL, NULL]|
+----------------------+