如果A列包含某些特定的字符串或A列中句子之外的單詞集，則如何更新B列的值

Question

如果A列中有句子：hb_mumbai煉油廠數據。 所以我需要檢查列A是否包含hb_mumbai，然后將列B的值更新為：hb_mumbai

我嘗試使用contains函數來實現它，但是沒有產生期望的輸出。

if df['A'].str.contains('hb_mumbai'): 
    df['B']=='hb_mumbai'

實際結果：B列的值未更新所需結果：B列的值應更新。

Answer 1

'=='是相等運算符，不分配值，請改用'='

Answer 2

如果df是字典，則可以執行以下操作：

if 'hb_mumbai' in df['A']:
    df['B']='hb_mumbai'

如前所述， ==用於檢查相等性； =將分配一個值。

Answer 3

import pandas as pd

df = pd.DataFrame({"A":['hb_mumbai in here','asd','qwe'],
                   "B":['123','456','789'] })

print(df)


df['B'] = ['hb_mumbai' if 'hb_mumbai' in a else b for a,b in zip(df['A'],df['A'])]

print(df)