带条件数据的数据框列 - Python

时间:2012-11-10 22:58:50

标签: python numpy dataframe pandas

我在数据框列中有一些百分比

pc
0.32
0.45
0.49
0.60
0.68
0.87

我想最终得到像这样的东西

pc     group
0.32    1
0.45    2
0.49    2
0.60    2
0.68    3
0.87    3

我试过

df["group"]=3

if df["pc"]<0.66:
    df["group"]=2

elif df["pc"]<0.33:
    df["group"]=1

但我得到的只是

ValueError: The truth value of an array with more than one element is ambiguous.

有什么想法吗?

1 个答案:

答案 0 :(得分:4)

df["group"][df["pc"] < 0.66] = 2
df["group"][df["pc"] < 0.33] = 1