在 Pandas dataframe 中循環 OR 條件

Question

我已經編寫了以下代碼塊來完成它需要做的事情（只要字符串“Hello”出現在 col1、col2、col3 或 col4 中，就保留該行）：

hello_mask = df["col1"].str.contains("Hello",na=False) | df["col2"].str.contains("Hello",na=False) | df["col3"].str.contains("Hello",na=False) | df["col4"].str.contains("Hello",na=False)
 
df_hello = df[hello_mask]

然而，它很丑陋，我認為 For 循環將是一個更優雅的解決方案（我可能是錯的）。 問題是我對coding不是最精通的，實在是拼不出來。

感謝您的耐心等待 - 任何建議都將不勝感激！

Answer 1

如果要檢查單元格是否完全匹配“Hello”，請使用：

cols_to_check = ['col1', 'col2', 'col3', 'col4']
hello_mask = df[cols_to_check].eq('Hello').any(axis=1)

如果要檢查部分字符串：

cols_to_check = ['col1', 'col2', 'col3', 'col4']
hello_mask = df[cols_to_check].apply(lambda x: x.str.contains('Hello')).any(axis=1)