数据帧的多个条件

Question

我正在尝试写一个新列'is_good'，如果'value'列中的数据集在范围1到6 之间，并且'value2'列在< strong>范围5到10 如果它们不满足两个条件，则标记为0

我知道你是否这样做，

df['is_good'] = [1 if (x >= 1 and x <= 6) else 0 for x in df['value']]

它将根据值的范围填充1或0，但在标记1或0时我还将如何考虑value2的范围。

无论如何我可以在没有numpy的情况下实现这个目标吗？

提前谢谢！

Answer 1

我认为需要&（按位和）需要加倍between和链条条件：

df = pd.DataFrame({'value':range(13),'value2':range(13)})
df['is_good'] =  (df['value'].between(1,6) & df['value2'].between(5,10)).astype(int)

或使用4个条件：

df['is_good'] =  ((df['value'] >= 1) & (df['value'] <= 6) & 
                   (df['value2'] >= 5) & (df['value'] <= 10)).astype(int) 


print (df)
    value  value2  is_good
0       0       0        0
1       1       1        0
2       2       2        0
3       3       3        0
4       4       4        0
5       5       5        1
6       6       6        1
7       7       7        0
8       8       8        0
9       9       9        0
10     10      10        0
11     11      11        0
12     12      12        0

Answer 2

更短的选择：

In [47]: df['is_good'] = df.eval("1<=value<=6 & 5<=value2<=10").astype(np.int8)

In [48]: df
Out[48]:
    value  value2  is_good
0       0       0        0
1       1       1        0
2       2       2        0
3       3       3        0
4       4       4        0
5       5       5        1
6       6       6        1
7       7       7        0
8       8       8        0
9       9       9        0
10     10      10        0
11     11      11        0
12     12      12        0