Как Pandas .loc может принимать три аргумента?
Я смотрю чей-то код и вот что они написали
from financetoolkit import Toolkit
API_KEY = "FINANCIAL_MODELING_PREP_API_KEY"
companies = Toolkit(["AAPL", "MSFT", "GOOGL", "AMZN"], api_key=API_KEY, start_date = "2005-01-01")
income_statement_growth = companies.get_income_statement(growth=True)
display(income_statement_growth.loc[:, "Revenue", :])
По сути, этот код возвращает величину дохода для пары компаний, начиная с 2005 года и по настоящее время.
Что меня смущает, так это income_statement_growth.loc[:, "Revenue", :]
почему три аргумента? Я не понимаю, что делает третье двоеточие в этом коде.
Во всей документации, о которой я читал .loc, говорится, что он принимает два аргумента: один для строки и один для столбца, поэтому я немного озадачен тем, как он может принимать три и какова функция третьего двоеточия.
Я не знаком с этим пакетом. Можете ли вы отредактировать свой вопрос, включив в него импорт и т. д.? Также было бы неплохо, если бы вы связали какую-нибудь документацию.
@ewokx Пожалуйста, не добавляйте просто ограничения кода в сообщение. Редактирование нарушило отступ.
Ответы 2
Возвращаемое значение companies.get_income_statement(growth=True) — это DataFrame pandas с мультииндексом. Столбцы индексируются по периоду ('2019', '2020' и т. д.), а строки индексируются по комбинации тикера компании и элемента данных (например, ('AAPT', 'Revenue')).
Вы можете получить доступ к одному элементу следующим образом:
print(income_statement_growth['2020'][('AAPL', 'Revenue')])
А чтобы выбрать 'Revenue' для всех тикеров и всех периодов, используйте .loc:
revenues = income_statement_growth.loc[:, 'Revenue', :]
Для простого фрейма данных вы обычно видите два аргумента для .loc[], но поскольку это мультииндекс, для этого требуется три аргумента.
Спасибо, Grismar, за объяснение того, что этот фрейм данных был многоиндексным! Теперь становится понятно, почему так написано.
«Вы можете получить доступ к одному элементу следующим образом:» — Почему бы не использовать .at? income_statement_growth.at[('AAPL', 'Revenue'), '2020']
Где это задокументировано? Я не знал об этом; самое близкое, что я знал, было IndexSlice, вот так: idx = pd.IndexSlice; income_statement_growth.loc[idx[:, 'Revenue'], :]
Я только что вспомнил аналогичный способ, который уже знал: income_statement_growth.xs('Revenue', level=1)
@Grismar, ты уверен, что индекс фрейма данных в вопросе имеет только два уровня? Откуда вы взяли эту информацию?
@Bill, я изменил код OP с помощью оператора импорта, получил бесплатный ключ API и запустил код.
@wjandrea, конечно, можно было бы использовать .at (и, вероятно, так и было бы). Я просто хотел показать очень простой пример доступа к элементу только с помощью базовых индексов, без каких-либо причудливых методов, и .at может показаться таким же волшебным, как .loc
Является ли индекс income_statement_growth мультииндексом? Если да, то сколько уровней у него? Если их три, этот код действителен:
income_statement_growth.loc[:, "Revenue", :]
При этом будут выбраны все строки с «Доходом» на втором уровне индекса.
Однако обратите внимание, что в документации написано
это сокращенное обозначение может привести к двусмысленности в целом.
Поэтому, по-видимому, использовать его нецелесообразно.
В этом случае я бы использовал следующее, как рекомендовал @wjandrea в комментарии к одному из других ответов:
idx = pd.IndexSlice
income_statement_growth.loc[idx[:, "Revenue", :]]
Это дает понять, что вы индексируете только строки (а не столбцы).
На мой взгляд, иногда Pandas слишком либерален.
Какой тип
income_statement_growth? (похоже, вы используетеfinancetoolkit, но вам следует указать это явно в вопросе)