Dataframe query result to string

Question

To get a plain string from a dataframe query, I use:

df = pd.DataFrame({
    "name": ["a", "b", "c"],
    "value": [1,2,3],
})

df.query("value == 1")['name'].tolist()[0]

# 'a'

Just wondering if this is the right way to process a pandas.core.series.Series object (i.e. whether there is a more Pythonic way of doing this).

jezrael · Accepted Answer · 2022-05-26 09:38:50Z

1

I think it is correct, if value exist.

If possible not exist here is next + iter trick:

out = next(iter(df.query("value == 1")['name']), 'no match')

out = next(iter(df.loc[df["value"] == 1, 'name']), 'no match')

answered May 26, 2022 at 9:38

jezrael

868k103 gold badges1.4k silver badges1.3k bronze badges

Sign up to request clarification or add additional context in comments.

1 Answer 1