检查熊猫数据框列中的特定值

Question

如果 B 的值为 1，我正在尝试检索 A 中的值。但下面的代码会引发错误“ValueError：系列的真值不明确。使用 a.empty, a.bool(), a.item()、a.any() 或 a.all()"

    A   B
0   a1  18
1   b1  25229
2   c2  2
3   d3  12
4   e4  1

代码：

for a,b in df7.iteritems():
  if (df7['b'] == 1):
    print (df7['a'])

Answer 1

你可以使用一个简单的比较，比如

import pandas as pd

df = pd.DataFrame({'A': ['a1', 'b1', 'c2', 'd3', 'e4'], 'B': [18, 25229, 2, 12, 1]})
print(df[df['B'] == 1]['A'])

产量

4    e4
Name: A, dtype: object

Answer 2

另一个简单的代码是

df7[df7['b'] == 1]['a']

将给出 a 列的数据框，其中 b=1。