下面这段代码是为了判断 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)
最后一行改成:
map(if_novel,movies['keywords'])