使用 Pandas df.loc

Question

我有一個 csv 文件的 DataFrame，該文件正在被 pandas 讀取。我想做的是使用 df.loc 添加一個新列，但僅當來自另一列（稱為“SKU”）的值以以下結尾時才將值插入該列“-RF”和“-NEW”。

我正在處理的代碼如下。 它有正在讀取的 csv 文件，“original_file”，然后是兩個 .loc 命令。 應該添加一個名為“Cond”的新列，我希望將“NEW”添加到“SKU”列中的值以“-NEW”結尾的那一列，並將“USED”添加到“SKU”以“-RF”結尾的新列。

AttributeError: 'list' object 沒有屬性 'str'

這是我一直在犯的錯誤。

original_file = pd.read_csv('wc-product-export-20-9-2022-1663680891149.csv')

original_file.loc[['SKU'].str.endswith("-RF"),'Cond'] = 'USED'
original_file.loc[['SKU'].str.endswith("-NEW"),'Cond'] = 'NEW'

Answer 1

希望這就是您要找的。 您可以使用應用 function 檢查 SKU 列中的每個單元格並相應地插入到新列中。

original_file['NEWCOL'] = original_file['SKU'].apply(
    lambda x: 'USED' if x.endswith('-RF') else 'NEW' if x.endswith('-NEW') else None)

Answer 2

當您按名稱調用列時嘗試添加 df

original_file.loc[original_file['SKU'].str.endswith("-RF"),'Cond'] = 'USED'