Поиск пары значений в DataFrame
Есть DataFrame:
col_1|col_2|info|
-----------------
val1 | val2| 33 |
val4 | val3| 32 |
И есть пара значений, информацию по которым надо извлечь (val4, val3).
Какой оптимальный запрос составить, чтобы получить информацию из колонки info?
Кроме как использовать дважды метод query() на ум ничего не приходит.
Ответы (2 шт):
Автор решения: MaxU
→ Ссылка
попробуйте так:
res = df.query("col_1 == 'val4' and col_2 == 'val3'")["info"]
пример:
In [81]: df
Out[81]:
col_1 col_2 info
0 val1 val2 11
1 val4 val3 22
In [82]: res = df.query("col_1 == 'val4' and col_2 == 'val3'")["info"]
In [83]: res
Out[83]:
1 22
Name: info, dtype: int64
Автор решения: CrazyElf
→ Ссылка
Можно и по-другому, примерно так (пишу по памяти):
res = df.loc[(df["col_1"] == 'val4') & (df["col_2"] == 'val3'),["info"]]