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

0 vote187 views2 answers
Go goose
Go
208 votes501 views4 answers
Docs .google com
Go
7 votes184 views2 answers
Google colab upload a file
Go
0 vote170 views1 answer
Onion omega location of files on sd
Go
1 vote192 views4 answers
Linux kill process by user
Go
0 vote182 views1 answer
How much google pays to software engineer
Go
0 vote198 views1 answer
Go ticker
Go
0 vote222 views1 answer
Oauth with google and vue
Go
0 vote203 views4 answers
Buffered channels in golang
Go
3 votes188 views1 answer
Go string to byte array
Go