如何使用掩碼變量將特定值分配給 pandas 中的列數據？

Question

我有一個 dataframe 水果，fruits_df。 最初我的代碼中有這個：

mask = ((fruits_df['SEASON']=='SUMMER') | (fruits_df['COLOR']'YELLOW'))
fruits_df['FRUIT'] = np.where(mask, 'SUMMER FRUITS', '')

現在，如果我想專門過濾具有甜味且屬於 Alphonso 品種的黃色夏季水果，並將它們分配到新列“MANGO”。 我該怎么做？ 我已經嘗試了下面的代碼，但它沒有按預期工作。

mask = ((fruits_df['SEASON']=='SUMMER') &
(fruits_df['TASTE']=='SWEET') &
(fruits_df['VARIETY']=='ALPHONSO') &
(fruits_df['FRUIT']==''))
fruits_df['FRUIT'] = np.where(mask, 'MANGO', fruits_df['FRUIT'])

我是 pandas 的初學者。任何幫助將不勝感激。 謝謝。

Answer 1

我認為，您缺少掩碼變量中 FRUIT 列的黃色條件； 所以將掩碼編輯成這樣；

mask = ((fruits_df['SEASON']=='SUMMER') & 
        (fruits_df['TASTE']=='SWEET') & 
        (fruits_df['VARIETY']=='ALPHONSO') & 
        (fruits_df['FRUIT']=='YELLOW'))       # edited here

或者也許是；

mask = ((fruits_df['SEASON']=='SUMMER') & 
        (fruits_df['TASTE']=='SWEET') & 
        (fruits_df['VARIETY']=='ALPHONSO'))