import pandas as pd
data = {'Name': ["Evil Nights", "Perfect Day", "Unknown Song", "Dead End", "Angel Love", "Home Sweet Home", "The Curse"]}
df = pd.DataFrame(data)
negative_keywords = ["Evil", "Night", "Problem", "Sorrow", "Dead", "Curse"]
positive_keywords = ["Amazing", "Angel", "Perfect", "Sunshine", "Home", "Live", "Friends"]
def categorize_song(name):
if any(word in name for word in negative_keywords):
return "Negative"
elif any(word in name for word in positive_keywords):
return "Positive"
else:
return "Neither"
df["Song Category"] = df["Name"].apply(categorize_song)
print(df)
Ouput: