Python - 每次 A 列等于 0 时从 B 列返回值

Question

每当 A 列等于 0 时，我想返回 B 列的值。我当前的代码:

for i in df:
    checkfornulls = df.ix[(df['A'][i] == 0).idxmin(),'B']
    if checkfornulls is None:
        print "Column A Pass!"
    else:
        print checkfornulls

如果我自己运行它:

checkfornulls = df.ix[(df['A'][i] == 0).idxmin(),'B']
    if checkfornulls is None:
        print "Column A Pass!"
    else:
        print checkfornulls

然后它带回它找到的第一个 0。有谁知道如何使 for 循环工作，以便返回 A 等于 0 的每个实例？

Answer 1

pandas 提供了大量方法来完成此任务。关于哪个是最好的、最 pythonic 的(不管那是什么意思)还是 pandantic(另一个虚构的词)是完全个人的决定。

使用查询
df.query('A == 0').B
或者
df.query('A == 0')['B']