如何在pyspark的spark数据框中汇总多个列? [英] How can I sum multiple columns in a spark dataframe in pyspark?

问题描述

我有一个要累加的列名列表

I've got a list of column names i want to sum

columns = ['col1','col2','col3']

我该如何添加这三个并将其放在新列中?(以一种自动方式，这样我就可以更改列列表并获得新结果)

How can i add the three and put it in a new column ? (in an automatic way, so that i can change the column list and have new results)

我想要结果的数据框:

col1   col2   col3   result
 1      2      3       6

谢谢！

推荐答案

尝试一下:

df = df.withColumn('result', sum(df[col] for col in df.columns))

df.columns 将是df中的列的列表.

df.columns will be list of columns from df.

这篇关于如何在pyspark的spark数据框中汇总多个列?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！