pandas中判断一个字符是否包含另一个字符,判断不出来

下面这段代码是为了判断 CSV 文件中的 'keywords' 列中的每个元素是否包含字符 'based on novel',然后输出 'if_novel' 列,但是运行之后,if 的情况并没有执行,全部是 else 的结果:

import pandas as pd

movies = pd.read_csv('movies.csv')

def if_novel(data):
    if "based on novel" in str(data):
        movies['if_novel'] = 'Based on novel'
    else:
        movies['if_novel'] = 'Not based on novel'
        
movies['keywords'].map(if_novel) 
阅读 7k
1 个回答

最后一行改成:
map(if_novel,movies['keywords'])

撰写回答
你尚未登录,登录后可以
  • 和开发者交流问题的细节
  • 关注并接收问题和回答的更新提醒
  • 参与内容的编辑和改进,让解决方法与时俱进