Как вытянуть значение из DataFrame?

Question

[Илья]

Хочу взять конкретное значение из DataFrame и назначить его переменной. Выглядит это примерно так

df = pd.DataFrame([[1000, 2], [1001, 5], [1002, 8]],
     columns=['client_id', 'segment'])

Хочу назначить переменной client_id значение из поля client_id, где поле segment равно какому-то значению. Нашел в документации такой вариант:

segment = 2
client_id = df.loc[df['segment'] == segment, 'client_id']

Возвращается Series с индексом. Как можно получить именно число 1000 из примера выше?

P.S. Значение client_id в DataFrame точно уникальное. Т.е. в финальной Series будет однозначно одна строка

Answer 1

[zexer]

client_id.values[0]

Answer 2

[Dmitry]

client_id = df.loc[df['segment'] == segment, 'client_id'][0]
# или
client_id = df.loc[df['segment'] == segment, 'client_id'].item()
# или
client_id = df[df['segment'] == segment]['client_id'].squeeze()

https://pandas.pydata.org/pandas-docs/stable/refer...