array_repeat

建立一個包含重複計數次數欄位的陣列。

語法

from pyspark.sql import functions as sf

sf.array_repeat(col, count)

參數

參數	類型	Description
col	pyspark.sql.Column 或 str	欄位名稱或表示要重複元素的表達式。
count	pyspark.sql.Column、str 或 int	欄位名稱、表達式或代表重複該元素次數的整數。

退貨

pyspark.sql.Column： 一個包含重複元素陣列的新欄位。

範例

範例 1：字串的使用

from pyspark.sql import functions as sf
df = spark.createDataFrame([('ab',)], ['data'])
df.select(sf.array_repeat(df.data, 3)).show()

+---------------------+
|array_repeat(data, 3)|
+---------------------+
|         [ab, ab, ab]|
+---------------------+

範例 2：整數的使用

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

+---------------------+
|array_repeat(data, 2)|
+---------------------+
|               [3, 3]|
+---------------------+

範例 3：陣列的使用

from pyspark.sql import functions as sf
df = spark.createDataFrame([(['apple', 'banana'],)], ['data'])
df.select(sf.array_repeat(df.data, 2)).show(truncate=False)

+----------------------------------+
|array_repeat(data, 2)             |
+----------------------------------+
|[[apple, banana], [apple, banana]]|
+----------------------------------+

範例 4：與 null 的使用

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

+---------------------+
|array_repeat(data, 3)|
+---------------------+
|   [NULL, NULL, NULL]|
+---------------------+