Pyspark udf multiple inputs

from pyspark.sql.types import IntegerType
from pyspark.sql.functions import udf

def sum(x, y):
    return x + y

sum_cols = udf(sum, IntegerType())

a=spark.createDataFrame([(101, 1, 16)], ['ID', 'A', 'B'])
a.withColumn('Result', sum_cols('A', 'B')).show()

Source: stackoverflow.com

Similar codes

213 votes411 views6 answers
Gamestop stocks
Go
1 vote180 views1 answer
Append to file in golang
Go
0 vote171 views1 answer
Perlin noise godot
Go
5 votes186 views2 answers
Cs go
Go
0 vote170 views1 answer
Run thread golang
Go
1 vote236 views1 answer
Kubernetes get pod crash logs
Go
7 votes167 views2 answers
Google colab upload a file
Go
5 votes195 views2 answers
Minecraft logo transparent
Go
2 votes172 views1 answer
Go declarations
Go
3 votes174 views2 answers
Delete all pods in default namespace
Go