根據條件從數據框中的現有列創建新列

Question

根據以下條件創建一個新列。 A 列的數據類型是字符串

如果 A 列字符串中存在文本。 創建僅包含字符串的 B 列。 例如： A 列：該產品用於管道安裝器 期望：如果 A 列包含管道安裝器 B 列 = 管道安裝器

"""df["New_text"] = []
def update_text(selected_text):
    for text in df['Activity Name']:
        if selected_text in text:
            df['New_text'] = df['New_text'].append('text')"""

Answer 1

所以如果我理解正確的話。 您的數據框中有一列Activity Name ，並且您想編寫一個方法，該方法每次都會打印具有特定關鍵字的數據框的子集。

如果我的理解是對的。

def update_text(df, selected_text):
    return df[df["Activity Name"].str.contains(selected_text)]