PySpark DataFrame 添加自增 ID

在用 Spark 处理数据的时候,经常需要给全量数据增加一列自增 ID 序

号,在存入数据库的时候,自增 ID 也常常是一个很关键的要素。

在 DataFrame 的 API 中没有实现这一功能,所以只能通过其他方式实

现,或者转成 RDD 再用 RDD 的 zipWithIndex 算子实现。

下面呢就介绍三种实现方式。

   创建 DataFrame 对象

  from pyspark.sql import SparkSession

  spark = SparkSession.builder.getOrCreate()

 df = spark.createDataFrame(
[
    {"name": "Alice", "age": 18},
    {"name": "Sitoi", "age": 22},
    {"name": "Shitao", "age": 22},
    {"name": "Tom", "age": 7},
    {"name": "De", "age": 17},
    {"name": "Apple", "age": 45}
]
)
df.show()

输出:

+---+------+
|age|  name|
+---+------+
| 18| Alice|
| 22| Sitoi|
| 22|Shitao|
|  7|   Tom|
| 17|    De|
| 45| Apple|
+---+------+

1**,monotonically_inc

  • 5
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值