cancel
Showing results forย 
Search instead forย 
Did you mean:ย 
Data Engineering
Join discussions on data engineering best practices, architectures, and optimization strategies within the Databricks Community. Exchange insights and solutions with fellow data engineers.
cancel
Showing results forย 
Search instead forย 
Did you mean:ย 

Trying to use pivot function with pyspark for count aggregate

rbricks007
New Contributor II

I'm trying this code but getting the following error

 

testDF = (eventsDF
          .groupBy("user_id")
          .pivot("event_name")
          .count("event_name"))

 

TypeError: _api() takes 1 positional argument but 2 were given

Please guide how to fix the error

1 REPLY 1

Krishnamatta
Contributor

Try this

from pyspark.sql import functions as F
testDF = (eventsDF
            .groupBy("user_id")
            .pivot("event_name")
            .agg(F.count("event_name")))