如何從 Excel 工作表中獲取兩個特定單詞並將它們添加到另一列 Python

Question

我是 Python 的新手，正在做一個項目，我需要清理我的數據。

我想添加一個只有“IN”或“OUT”字樣的新列。

完整數據的圖片，其中單詞“IN”和“OUT”是

import pandas as pd 
df = pd.read_excel("tourniquets_26.07.2022_test.xls") #Read Excel as file as Dataframe

#My try to use a loop to find it
for word in df["Deur intelligente eenheid"]:
    if word == "IN" or word == "UIT":
        df['IN/UIT'] = word
#Display top 5 rows  
print(df.head(5))

#To save it back as Excel
df.to_excel("tourniquets_26.07.2022_test.xls") #Write DataFrame back as Excel file

先感謝您

Answer 1

據我了解，邏輯是您當前df中的列總是（其中某處）單詞“IN”我們的“UIT”，並且您想要一個只有相應“IN”或“ UIT”這個詞，對吧？

如果是這樣，您可以使用Series.apply來執行此操作。 創建一個 function 根據句子中的單詞返回“IN”或“UIT”：

def my_fun(sentence):
    if "IN" in sentence:
        return "IN"
   else: #assuming that if it doesn't contain "IN", it must contain "UIT"
        return "UIT"

並將其應用於 df["col_1"] 以創建一個新列。 假設您的列名為“col_1”：

df['IN/UIT'] = df["col_1"].apply(my_fun)

Answer 2

您可以使用 np.where：

import numpy as np
df['IN/UIT'] = np.where(df['col'].str.startswith('IN'), 'IN', 'UIT')