Панды - как использовать значение из переменной (вместо имени переменной), как имя столбца при фильтрации? - PullRequest
0 голосов
/ 21 мая 2018

Как заставить панд использовать значение из переменной, а не само имя переменной при фильтрации?

Дано:

column_name = "weird_name"
df = pd.DataFrame(data={column_name: [1, 2, 3, 4, 5]})

возвращает:

    weird_name
0   1
1   2
2   3
3   4
4   5

сейчасЯ хотел бы отфильтровать странный столбец, однако обычный способ обрабатывает переменную column_name как имя столбца:

df[df.column_name > 3]

, поэтому очевидно, что он выдает:

AttributeError: 'DataFrame' object has no attribute 'column_name'

PS.Я был бы рад, если бы носитель английского языка отредактировал название, чтобы оно звучало более естественно.

1 Ответ

0 голосов
/ 21 мая 2018

РЕДАКТИРОВАТЬ: Как @jpp указал в комментариях, это правильный ответ:

df[df[column_name] > 3]
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...