如何將 if 條件應用於兩個不同的列並將結果放入新列

Question

我有一個數據框df2 ，並希望根據兩個現有列的if邏輯生成一個名為“標簽”的新列。

import pandas as pd
df2 = pd.DataFrame({'NOTES': ["PREPAID_HOME_SCREEN_MAMO","SCREEN_MAMO",
                              "> Unable to connect internet>4G Compatible>Set",
                              "No>Not Barred>Active>No>Available>Others>",
                              "Internet Not Working>>>Unable To Connect To"], 
     'col_1': ["voice", "voice","data","other","voice"],
     'col_2': ["DATA", "voice","VOICE","VOICE","voice"]})

邏輯和我的嘗試是：

df2['Tag'] =             
            if df['col_1']=='data':
                return "Yes"
            elif df['col_2']:
                return "Yes"
            else:
                return "No"

但是我遇到語法錯誤：

Answer 1

問題是您試圖用 if-語句賦值，這會導致語法錯誤。

有很多方法可以做到這一點，我提供了一種使用pandas.DataFrame.apply的方法。

trans_fn = lambda row: "Yes" if row['col_1']=='data' && row['col_2'] else "No"
df2['tag'] = df2.apply(trans_fn, axis=1) # apply trans_fn to each row