python pandas如果列字符串包含單詞標志

Question

我想在我的python pandas dataframe df中添加一個標志，如果Title列中的條目包含單詞test （大寫或小寫或全部大寫），我想在新的列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