在python中创建一个新变量，其值应该是imdbrating和imdbvotes之间的差的平方

Question

这是我用来计算imdbRating和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']

这是我在Python 3.7.0+0.23.4中遇到的错误：

TypeError: string indices must be integers

（imdb_data是数据帧，引用的列名确实存在）

Answer 1

列imdbRating, imdbVotes应该是数据类型float。所以把它们从string转换成float。然后做你的计算。你知道吗

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()

输出为

    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