首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >PySpark -对数据格式中的列进行求和,并以int形式返回结果

PySpark -对数据格式中的列进行求和,并以int形式返回结果
EN

Stack Overflow用户
提问于 2017-12-14 11:43:05
回答 9查看 156.2K关注 0票数 56

我有一列数字的电火花数据。我需要对该列进行求和,然后在python变量中将结果作为int返回。

代码语言:javascript
运行
复制
df = spark.createDataFrame([("A", 20), ("B", 30), ("D", 80)],["Letter", "Number"])

我做下面的总结这一栏。

代码语言:javascript
运行
复制
df.groupBy().sum()

但我得到了一份数据。

代码语言:javascript
运行
复制
+-----------+
|sum(Number)|
+-----------+
|        130|
+-----------+

我会把130作为一个int返回到一个变量中,以便在程序中的其他地方使用。

代码语言:javascript
运行
复制
result = 130
EN

Stack Overflow用户

发布于 2018-10-28 07:18:40

这是另一种你能做到的方法。使用aggcollect

代码语言:javascript
运行
复制
sum_number = df.agg({"Number":"sum"}).collect()[0]

result = sum_number["sum(Number)"]
票数 16
EN
查看全部 9 条回答
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/47812526

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档