Creating array column but getting error : TypeError: 'list' object is not callable in PySpark

Question

I want to create a array column from existing column in PySpark

--------------------------
col0 | col1 | col2 | col3
--------------------------
1    |a     |b     |c
--------------------------
2    |d     |e     |f
--------------------------

I want like this

-------------
col0 | col1 
-------------
1    |[a,b,c]
-------------
2    |[d,e,f]
--------------

I was trying array() function like this

>>> new = df.select("col0",array("col1","col2","col3").alias("col1"))

but getting this error

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: 'list' object is not callable

Please if anyone have solution on this ..

It Worked after I restart my pyspark..

Onkar Jadhav
– Onkar Jadhav

2020-11-12 06:10:19 +00:00
Commented Nov 12, 2020 at 6:10 — Onkar Jadhav
– Onkar Jadhav, Commented Nov 12, 2020 at 6:10

dsk · Accepted Answer · 2020-11-09 11:43:47Z

1

You need to use withColumn() first while creating a new column , post that you can use select() in order to select columns as per your choice

df = df.withColumn("col0", array("col1","col2","col3"))
df = df.select("col0")

and you are getting this error because, you are using .alias() function and the compiler is complaining about that

answered Nov 9, 2020 at 11:43

dsk

2,0272 gold badges12 silver badges13 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

mck Over a year ago

alias should work on array because array returns a column. I suspect array was not pyspark.sql.functions.array but something else, but after restarting spark as in the answer below, it somehow got replaced by the correct spark array function.

Onkar Jadhav Over a year ago

Dont know .. I also tried with list() instead of array() , it gave me the same error

Collectives™ on Stack Overflow

Creating array column but getting error : TypeError: 'list' object is not callable in PySpark

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related