pyspark.sql.functions.array_repeat

pyspark.sql.functions.array_repeat(col, count)[source]

Collection function: creates an array containing a column repeated count times.

New in version 2.4.0.

Examples

>>> df = spark.createDataFrame([('ab',)], ['data'])
>>> df.select(array_repeat(df.data, 3).alias('r')).collect()
[Row(r=['ab', 'ab', 'ab'])]