我正在尝试用python编写一些代码,如果单元格值包含某个单词,这些代码将拆分单元格中的字符串。
我制作了一个示例数据帧来解释我所面临的挑战。
d = {'Message': [1, 2,3], 'Details': ['I WANT IT ALL', 'HELLO WORLD','NOPE IT IS NOT CORRECT']}
df = pd.DataFrame(data=d)
df
我想不出这个代码的其余部分。
d['New Column'] = pd.np.where(d['Details'].str.contains("WANT"),
如果列包含";想要";"New Column"值="want all"ELSE other。
提前感谢社区。
到此为止。
def get_left(value: str) -> str:
return df.query("Details.str.contains(@value)").at[0, "Details"].split(value)[0]
df["New Column"] = np.where(df["Details"].str.contains("WANT"), get_left("WANT"), "other")
print(df)
Message Details New Column
0 1 I WANT IT ALL I
1 2 HELLO WORLD other
2 3 NOPE IT IS NOT CORRECT other