python pandas for循環向新列添加值

Question

我想為 python pandas 中的每個循環添加一個新值

        PRODUCT  PRICE
0    ABC       5000
1    ABB       2500
2    ABE       1800
3    DCB       7200

如果 df1 像上面那樣，

for i in df1['PRICE']:
    if i>3000:
        PURCHASE = True
    else:
        PURCHESE = False

循環后，我想使 df1 如下所示

        PRODUCT  PRICE  PURCHSE
0    ABC       5000       True
1    ABB       2500       False
2    ABE       1800       False
3    DCB       7200       True

我該怎么做？

Answer 1

Pandas 的語言有點類似於您在普通對話中使用的語言：PURCHASE 是當 PRICE > 3000 時。

df1['PURCHASE'] = df1['PRICE'] > 3000

Answer 2

您能否嘗試在此處使用np.where進行以下操作。

import pandas as pd
import numpy as np
df['PURCHASE']=np.where(df2.PRICE>3000,'TRUE','FALSE')

當我們使用名為PURCHASE的新列打印 new df的值時，其值將如下所示。

    PRODUCT PRICE PURCHASE
0   ABC 5000    TRUE
1   ABB 2500    FALSE
2   ABE 1800    FALSE
3   DCB 7200    TRUE