根據 if 和現有列在 pandas 中創建新列

Question

我找到了這個問題的不同答案，但沒有從現有列中提取數據。 假設我有 DataFrame

purchase=
{'date':['11/03/2021','12/03/2021','14/03/2021','11/03/2021'],
'price':[300, 400,200, 200],
'currency':['eur', 'usd','usd','usd'],
'qty':[200, 300, 400, 500],
'salesmanA':['Jack', 'x', "Mike", 'x'],
'salesmanB':['x', 'John', "x", 'David']}
df=pd.DataFrame(purchase)

我想設置一個新列 df['salessup'] 如果它的值不是'x'，它應該等於 SalesmanA，如果它是 x 到 salesmanB 新列應該像 ['salessup']=['Jack', 'John','Mike','David'] 提前謝謝你。

Answer 1

試試np.where

df['out'] = np.where(df.salesmanA=='x',df.salesmanB,df.salesmanA)
df
Out[450]: 
         date  price currency  qty salesmanA salesmanB    out
0  11/03/2021    300      eur  200      Jack         x   Jack
1  12/03/2021    400      usd  300         x      John   John
2  14/03/2021    200      usd  400      Mike         x   Mike
3  11/03/2021    200      usd  500         x     David  David