Python scatter plots for aggregated columns in a pandas dataframe

Question

I have a dataset of every player in the NBA and their stats since 1950. The columns in the dataset consist of the Year, which is the applicable year, player names and his team for that year, years in the NBA, and 20 columns of different stats for every player in every year of his career. One of the columns is 'PTS', which is the total number of points the player has scored that year. I want to create a scatter plot in Python that shows the Years 1950 through 2017 on the x-axis and the total points scored in that year on the y-axis. I believe the most efficient way to return the total points for each season is:

    import numpy as np
    import pandas as pd
    import matplotlib.pyplot as plt
    import seaborn as sns
    seasons = pd.read_csv('C:/windows/temp/Seasons_Stats.csv')
    tp_yr = seasons.groupby('Year').agg({'PTS': ['sum']})

But creating the scatter plot using:

    tp_yr.plot.scatter( x= 'Year', y = 'PTS', s = 'None', c='red')

returns:

KeyError: 'Year'

and a blank graph I want the total points in a year for all the years from 1950 -2017 represented in a red scatter plot.

"I'm having issues" is not a sufficient problem description. Be specific: What are the columns in the dataframe? What should the final plot show (what is the xaxis, what is the yaxis, what should the points represent?), what have you tried to create such a plot? What is the problem? In how far does it not match your expectations, or do you get an error? As you can see the list of open questions is longer than your question itself. Please edit it to make it answerable. — ImportanceOfBeingErnest
– ImportanceOfBeingErnest, Commented Jun 12, 2017 at 16:16
Converting the "Year" column would solve the Error that you've posted. (See: stackoverflow.com/questions/35432918/…) To answer the other questions, you need to show a bit more of your code. — mimre
– mimre, Commented Jun 12, 2017 at 17:27
Thanks, I have shown the rest of my code with a few updates from info I found on plotting pandas Dataframes, but it is still not working like I need, suggestions? — Robert Dwyer
– Robert Dwyer, Commented Jun 12, 2017 at 23:51

ImportanceOfBeingErnest · Accepted Answer · 2017-06-13 10:06:09Z

2

You're getting a key error, because there is no column named "Year" in the aggregated dataframe, since the year is used as index.

In order to get back the index as a column of the dataframe use .reset_index().

Something like this should work:

ptsbyyear = df.groupby("Year").agg({'PTS': ['sum']}).reset_index()
ptsbyyear.plot(kind="scatter", x="Year", y="PTS")

answered Jun 13, 2017 at 10:06

ImportanceOfBeingErnest

342k61 gold badges739 silver badges771 bronze badges

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

1 Comment

Robert Dwyer Over a year ago

Thanks so much. I noticed that the year want a column name and realized that was the problem but didn't know how to retrieve the year as a . Thanks again.

Collectives™ on Stack Overflow

Python scatter plots for aggregated columns in a pandas dataframe

1 Answer 1

1 Comment

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Linked

Related