在 Dataframe Python 中放置一個新列

Question

我正在處理的 Dataframe 有一個名為“Brand”的列，該列的值名為“SEAT”，帶有空白。 我實現了刪除空白，但我不知道如何將新列放入之前的 Dataframe 中。我必須這樣做，因為我需要通過“SEAT”過濾之前的 Dataframe 並顯示這些行。

我試過這個：

import pandas as pd

brand_reviewed = df_csv2.Brand.str.rstrip()

brand_correct = 'SEAT'

brand_reviewed.loc[brand_reviewed['Brand'].isin(brand_correct)]

非常感謝你！

Answer 1

據我所理解，

您正在嘗試返回與模式“SEAT”匹配的行。 您不會被迫創建新列。 您可以直接執行以下操作：

df2 = brand_reviewed[brand_reviewed.Brand.str.rstrip() == "SEAT"]
print(df2)

Answer 2

你做得很好。 我還將提到另一種形式，說明如何清除空白。 而且，如果您只想在當前的 dataframe 中添加一個新列，只需編寫此代碼的最后一行。

import pandas as pd

brand_reviewed = pd.read_csv("df_csv2.csv")

data2 = data["Brand"].str.strip()

brand_reviewed["New Column"] = data2

如果您有其他疑問，請告訴我。

奧克塔維奧·委拉斯開茲