Create new variable whose values should be square of difference between imdbrating and imdbvotes in python

Question

Here is my code to calculate the square difference between imdbRating and imdbVotes

imdb_data['imdbVotes'] = imdb_data['imdbVotes'].astype(int)  
imdb_data['imdbRating'] = imdb_data['imdbRating'].astype(int)

imdb_data['new'] = imdb_data['imdbRating'] - imdb_data['imdbVotes']

This is the error I got with Python 3.7.0 + pandas 0.23.4:

TypeError: string indices must be integers

(imdb_data is a dataframe and the column-names referenced do exist)

`imdb_data` is not a dataframe if that's the error you're getting? — cs95, Dec 17 '19 at 05:22
no that is not the error . imdb_data i converted it into thr required datattypes . I guess its because of the missing values in the dataset or its not cleaned . — Amith Gowda, Dec 17 '19 at 05:26
Can you provide sample dataframe with values, so that others can copy paste and check your problem ? — Prince Francis, Dec 17 '19 at 05:26
here is the link to the entire dataset (https://drive.google.com/open?id=1KvWzmqLaA_ZlKoKRSz0V1ycm4FIOdICQ) @PrinceFrancis — Amith Gowda, Dec 17 '19 at 05:33

Prince Francis · Accepted Answer · 2019-12-17T06:42:55.640

2

The columns imdbRating, imdbVotes should be of datatype float. So convert them to float from string. Then do your calculations.

imdb_data = pd.read_csv('IMDB_data.csv', sep=',',encoding = 'ISO-8859-1')
imdb_data['imdbRating'] = pd.to_numeric(imdb_data['imdbRating'], errors='coerce', downcast='float')
imdb_data['imdbVotes'] = pd.to_numeric(imdb_data['imdbVotes'], errors='coerce', downcast='float')
imdb_data['new'] = imdb_data['imdbRating'] - imdb_data['imdbVotes']
imdb_data.head()

output is

    Plot    Title   imdbVotes   Poster  imdbRating  Genre   imdbID  Year    Language    new
0   Despite his tarnished reputation after the eve...   The Dark Knight Rises   2679.0  http://ia.media-imdb.com/images/M/MV5BMTk4ODQz...   75.0    Action, Thriller    tt1345836   2012    English     -2604.0
1   0   0   0.0     0   0.0     0   0   0   0   0.0
2   Based on the novel written by Stephen Chbosky,...   The Perks of Being a Wallflower     1270.0  http://ia.media-imdb.com/images/M/MV5BMzIxOTQy...   71.0    Drama, Romance  tt1659337   2012    English     -1199.0
3   Mike Lane is a thirty-year old living in Tampa...   Magic Mike  2580.0  http://ia.media-imdb.com/images/M/MV5BMTQzMDMz...   51.0    Comedy, Drama   tt1915581   2012    English     -2529.0
4   When Bond's latest assignment goes gravely wro...   Skyfall     1807.0  http://ia.media-imdb.com/images/M/MV5BMjAyODkz...   68.0    Action, Thriller    tt1074638   2012    English     -1739.0

edited Dec 17 '19 at 06:42

answered Dec 17 '19 at 05:44

Prince Francis

2,995
1
14
22

I tried the above code and still encountered the same error , also tried this line of code " imdb_data['imdbRating'] = imdb_data['imdbRating'].astype(float)" which is also getting the same error saying string indices must be integers @PrinceFrancis – Amith Gowda Dec 17 '19 at 06:07
What is your python and pandas version ? – Prince Francis Dec 17 '19 at 06:17
Python 3.7.0 and pandas: 0.23.4 – Amith Gowda Dec 17 '19 at 06:31
1

I am using pandas `0.25.3` and python `3.7.5`. It works here. I edited my answer with output. – Prince Francis Dec 17 '19 at 06:41
1

updated the pandas version and now it works – Amith Gowda Dec 17 '19 at 07:08
but when i tried to get the square of the 'new' column . I'm getting the error saying series not callable – Amith Gowda Dec 17 '19 at 07:22
how you tried ? Use like these `imdb_data['square'] = imdb_data['new']**2` – Prince Francis Dec 17 '19 at 07:24

Create new variable whose values should be square of difference between imdbrating and imdbvotes in python

1 Answers1