首页 > 解决方案 > Python中的错误:列表索引超出范围

问题描述

def item(id):
return posts.loc[posts['post_id'] == id]['title'].tolist()[0]

def get_recommendations(postid, num, indices):
idx = indices[postid]
sim_scores = list(enumerate(cosine_similarities[idx]))
sim_scores = sorted(sim_scores, key=lambda x: x[1], reverse=True)
sim_scores = sim_scores[1:num+1]
# print(sim_scores)
indices = [i[0] for i in sim_scores]
print("Recommending " + str(num) + " posts similar to \"" + item(postid) + "\" ...")
return posts.iloc[indices]

这是我的代码,当我运行它时,出现以下错误:

IndexError                                Traceback (most recent call last)
<ipython-input-58-54719cff7892> in <module>()
----> 1 show_recomm('5d6d39567fa40e1417a4931c', 10, indices)

1 frames
<ipython-input-45-dfd12eb7d387> in show_recomm(postid, num, indices)
      6     print(similarity_scores)
      7     indices = [i[0] for i in similarity_scores]
----> 8     print("Recommending " + str(num) + " posts similar to \"" + item(postid) + "\" ...")
      9     return posts.iloc[indices, :-1]

<ipython-input-44-6739fecf9da1> in item(id)
      1 def item(id):
----> 2     return posts.loc[data['post_id'] == id]['title'].tolist()[0]

IndexError: list index out of range

当我尝试使用不同帖子 ID 的函数时出现此错误。对于某些人,我得到了输出,而对于其他人,我没有。

标签: pythonpython-3.xrecommendation-enginerecommender-systems

解决方案


根据我在上面看到的。似乎是由于

data['post_id'] == id

这将是真或假。你的意思

def item(id):
    return posts.loc[id]['title'].tolist()[0]

推荐阅读