pyspark.sql.functions.array

pyspark.sql.functions.array(*cols)[source]

Creates a new array column.

New in version 1.4.0.

Parameters
colsColumn or str

column names or Columns that have the same data type.

Examples

>>> df.select(array('age', 'age').alias("arr")).collect()
[Row(arr=[2, 2]), Row(arr=[5, 5])]
>>> df.select(array([df.age, df.age]).alias("arr")).collect()
[Row(arr=[2, 2]), Row(arr=[5, 5])]