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 vote336 views1 answer
Clear all children godot
Go
1 vote344 views1 answer
Golang backward loop
Go
0 vote350 views1 answer
{google:baseURL}search?q=%s&{google:RLZ}{google:originalQueryForSuggestion}{google:assistedQueryStats}{google:searchFieldtrialParameter}{google:iOSSearchLanguage}{google:searchClient}{google:sourceId}{google:contextualSearchVersion}ie={inputEncod
Go
0 vote309 views1 answer
Golang writefile example
Go
0 vote312 views1 answer
Golang passing parameter to the program
Go
0 vote338 views1 answer
Gofiber basic routing
Go
0 vote316 views1 answer
Golang byte slice to reader
Go
0 vote345 views1 answer
Excelize
Go
0 vote307 views2 answers
Google chrome path in mac
Go
1 vote328 views3 answers
Interface to int golang
Go