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
New Contributor III

Try this

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

 

 

Join Us as a Local Community Builder!

Passionate about hosting events and connecting people? Help us grow a vibrant local community—sign up today to get started!

Sign Up Now