时间戳

首页 > 解决方案 > 您如何获得通过数据框中的值搜索的特定值？

python - 您如何获得通过数据框中的值搜索的特定值？

问题描述

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

名称：计数，数据类型：int64

标签： pythonpandasdataframe

解决方案

df.loc返回 a 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]

推荐阅读