从数据框中选择特定值

Question

我正在尝试从数据框中选择一个值。但问题是输出带有数据类型和列名。这是我从 csv 文件中读取的数据框，

Name,Code
blackberry,1
wineberry,2
rasberry,1
blueberry,1
mulberry,2

这是我的测试代码-

dataFrame=pd.read_csv("test.csv")
value = dataFrame.loc[dataFrame['Name'] == 'rasberry']['Code']
print(value)
strvalue=str(value)
if(strvalue=="1"):
    print("got it")

的预期输出value将是1但它是

2  1\nName: Code, dtype: int64

这就是 if 条件不起作用的原因。我怎样才能得到具体的价值？我在用pandas

Answer 1

你得到的值是一个 Series 对象。您可以使用.iloc它从中提取值：

value.iloc[0]
# 1

或者您可以使用.values提取底层 numpy 数组，然后使用索引提取值：

value.values[0]
# 1