我正在尝试从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
答案 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]