python pandas 如果列字符串包含单词标志

Question

如果 Title 列中的条目包含单词 test(大写或小写或全部大写)，我想向我的 python pandas 数据帧 df 添加一个标志，我想在新列 test 中添加 T。

这给了我一个错误，并没有考虑到所有大写的情况:

df['Test_Flag'] = np.where(df[df['Title'].str.contains("test|Test")==True], 'T', '')

ValueError: Length of values does not match length of index

Answer 1

你需要contains带参数 case=False 和 na=False:

df['Test_Flag'] = np.where(df['Title'].str.contains("test", case=False, na=False), 'T', '')

示例:

df = pd.DataFrame({'Title':['test','Test',np.nan, 'a']})
df['Test_Flag'] = np.where(df['Title'].str.contains("test", case=False, na=False), 'T', '')
print (df)
  Title Test_Flag
0  test         T
1  Test         T
2   NaN          
3     a