當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


Python pyspark RDD.take用法及代碼示例


本文簡要介紹 pyspark.RDD.take 的用法。

用法:

RDD.take(num)

取 RDD 的前 num 個元素。

它首先掃描一個分區,然後使用該分區的結果來估計滿足限製所需的附加分區數。

翻譯自 RDD#take() 中的 Scala 實現。

注意

僅當預期結果數組很小時才應使用此方法,因為所有數據都加載到驅動程序的內存中。

例子

>>> sc.parallelize([2, 3, 4, 5, 6]).cache().take(2)
[2, 3]
>>> sc.parallelize([2, 3, 4, 5, 6]).take(10)
[2, 3, 4, 5, 6]
>>> sc.parallelize(range(100), 100).filter(lambda x: x > 90).take(3)
[91, 92, 93]

相關用法


注:本文由純淨天空篩選整理自spark.apache.org大神的英文原創作品 pyspark.RDD.take。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。