檢查熊貓數據框列中的特定值

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。