Python pyspark range用法及代碼示例

創建一個包含一定範圍數字的DataFrame。

生成的 DataFrame 有一個名為 id 的 int64 列，包含從 start 到 end(不包括)範圍內的元素，步長 值為 step 。如果隻指定了第一個參數(即start)，我們將其視為結束值，起始值為0。

這類似於SparkSession中的範圍函數，主要用於測試。

參數：

start：int

起始值(含)

end：整數，可選

最終值(不包括)

step：int，可選，默認1

增量步驟

num_partitions：整數，可選

DataFrame的分區數量

返回：

DataFrame

例子：

當指定第一個參數時，我們會生成一個範圍內的值，直到該數字為止。

>>> ps.range(5)
   id
0   0
1   1
2   2
3   3
4   4

指定 start、end 和 step 時：

>>> ps.range(start = 100, end = 200, step = 20)
    id
0  100
1  120
2  140
3  160
4  180