python pandas for循环将值添加到新列

Question

        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

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

我该怎么做呢？

Answer 1

Pandas的语言有点类似于你在普通对话中使用的语言:当价格>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的新列打印新df的值时，它的值将如下所示。

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