如何获取数据框中的值搜索的特定值?

时间:2019-04-11 04:17:04

标签: python pandas dataframe

我正在尝试从DataFrame中获取特定值。但是我想通过一个值来搜索该值。

我一直在尝试df.loc [df ['word'] =='apple','count']。

这给了我索引和计数。我想要的只是计数,这样我就可以将其用作数字来计算周围的事物。

d = {'word' : ['apple', 'banana'], 'count' : [1,2] }
df = pd.DataFrame(data=d)

df.loc[df['word']=='apple','count']

我期望或希望输出为1

,类型为整数或浮点型, 但输出是:

0 1

名称:count,dtype:int64

1 个答案:

答案 0 :(得分:1)

df.loc返回pandas.Series,即使该系列中只有一个元素也是如此。从系列中获取原始值的一种方法是在单个位置对其进行索引。将.iloc[0]添加到您的代码中会产生所需的结果:

import pandas as pd

d = {'word': ['apple', 'banana'], 'count': [1, 2]}
df = pd.DataFrame(d)
df.loc[df['word'] == 'apple', 'count'].iloc[0]