Как вы получаете конкретное значение, которое ищется по значению в кадрах данных?

Question

Я пытаюсь получить конкретное значение из 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

Answer 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]