topic TypeError : withcolumn() takes 3 positional arguments but 4 were given. in Data Engineering

TypeError : withcolumn() takes 3 positional arguments but 4 were given.

JohnJustus — Thu, 21 Sep 2023 18:48:56 GMT

Hi All,

Can some one please help me with the error.

This is my small python code.

binmaster = binmasterdf.withColumnRenamed("tag_number","BinKey")\

.withColumn ("Category", when (length("Bin")==4,'SMALL LOT'),(length("Bin")==7,'RACKING'))

TypeError : withcolumn() takes 3 positional arguments but 4 were given.

If I remove this (length("Bin")==7,'RACKING'), there are no issues.

Can some one please let me know how to fix the error?

Thanks,

John

Re: TypeError : withcolumn() takes 3 positional arguments but 4 were given.

JohnJustus — Fri, 22 Sep 2023 14:25:24 GMT

Thanks for your response

Actually, here is my code with when function but I still see the error

binmaster = binmasterdf.withColumnRenamed("tag_number","BinKey")\

.withColumn ("Category", when (length("Bin")==4,'SMALL LOT'), when (length("Bin")==7,'RACKING'), otherwise('FLOOR'))

Is there anything wrong with the above?

Thanks,

John

Re: TypeError : withcolumn() takes 3 positional arguments but 4 were given.

Noopur_Nigam — Tue, 26 Sep 2023 11:22:17 GMT

Hi @JohnJustus If you see closely in .withColumn ("Category", when (length("Bin")==4,'SMALL LOT'), when (length("Bin")==7,'RACKING'), otherwise('FLOOR')), withcolumn would take 2 parameters. The first parameter as a string and the second as the column. However in your code, you are passing 3 parameters i.e. first is string and the other two are when conditions so the last when condition is being considered as an extra parameter in withcolumn function. This is the reason removing last when is resolving the issue.

Re: TypeError : withcolumn() takes 3 positional arguments but 4 were given.

JohnJustus — Tue, 26 Sep 2023 14:25:12 GMT

Thank you Noopur for your response.

So how can I accomplish with more than 2 parameters? Are there any other function available?

In this case, I have to check length multiple times and categorize them accordingly.

Thanks,

John