Question

鉴于此DataFrame：

df = pandas.DataFrame({"a": [1,10,20,3,10], "b": [50,60,55,0,0], "c": [1,30,1,0,0]})

制作新专栏的最佳方式是什么，＆＃34;过滤＆＃34;有价值的＆＃34;传递＆＃34;如果列a和b的值都大于x且值为＆＃34;则失败＆＃34;否则？

可以通过遍历行来完成，但它效率低且不够优雅：

c = []

for x, v in df.iterrows():
     if v["a"] >= 20 and v["b"] >= 20:
         c.append("pass")
     else:
         c.append("fail")

df["filter"] = c

Answer 1

一种方法是创建一个布尔值列，如下所示：

>>> df['filter'] = (df['a'] >= 20) & (df['b'] >= 20)
    a   b   c filter
0   1  50   1  False
1  10  60  30  False
2  20  55   1   True
3   3   0   0  False
4  10   0   0  False

然后，您可以使用replace将布尔值更改为“通过”或“失败”：

>>> df['filter'].astype(object).replace({False: 'fail', True: 'pass'})
0    fail
1    fail
2    pass
3    fail
4    fail

您可以使用all将此扩展到更多列。例如，要查找条目大于0的列中的行：

>>> cols = ['a', 'b', 'c'] # a list of columns to test
>>> df[cols] > 0 
      a      b      c
0  True   True   True
1  True   True   True
2  True   True   True
3  True  False  False
4  True  False  False

在此DataFrame的第1轴上使用all会创建新列：

>>> (df[cols] > 0).all(axis=1)
0     True
1     True
2     True
3    False
4    False
dtype: bool

基于过滤器在pandas DataFrame中创建新列

1 个答案: