Criando estratégias de investimento com IA no Python

Imagine ter a capacidade de pedir à inteligência artificial para criar uma estratégia de investimento baseada em indicadores técnicos ou regras específicas. Com o tempo sendo um recurso valioso, nem sempre é possível desenvolver um código por conta própria. Vamos mostrar como a IA, junto com Python, pode facilitar a criação de estratégias de investimento.

Aprenda a coletar, processar e analisar dados na formação de Do Zero à Análise de Dados Econômicos e Financeiros com Python.

Objetivo

O Objetivo deste exercício é bem simples: vamos utilizar o Gemini para criar uma estratégia de investimentos por meio de um prompt. Vamos utilizar como exemplo as ações da PETR4 em frequência diária.

Prompt

Para obter os valores da estratégia em um arquivo CSV, usamos o seguinte prompt.

Você é um analista quantitativo focado em desenvolver estratégias de trading algorítmico.

Seu objetivo é criar uma estratégia utilizando o indicador MACD (Moving Average Convergence Divergence) para a ação PETR4.

Para isso, você deve:

1. Carregar o arquivo `petr4.csv`, que contém os dados históricos de preços da ação PETR4.

2. Calcular o MACD usando os parâmetros padrão (períodos de 12 e 26 para médias móveis exponenciais, e período 9 para a linha de sinal).

3. Gerar um arquivo CSV em formato de tabela que inclua as colunas:

- `Date`: Data da cotação

- `Close`: Cotação no período

- `MACD`: A diferença entre a média móvel exponencial de 12 períodos e a de 26 períodos.

- `Signal`: A média móvel exponencial de 9 períodos do MACD.

- `Histogram`: A diferença entre o MACD e a linha de sinal.

- `Sign`: coluna para referência com os valores de `Compra` ou `Venda`, conforme a lógica da estratégia.

- `Return`: Retorno Acumulado da estratégia baseado no sinal de compra e venda.

4. Salve o arquivo CSV com o nome `resultado.csv` e não retorne nada mais.

Saiba mais sobre prompts neste artigo sobre engenharia de prompt.

Visualização de dados

Para analisar o resultado da estratégia, criamos um código para visualizar os dados da estratégia:

Quer aprender mais?

Clique aqui para fazer seu cadastro no Boletim AM e baixar o código que produziu este exercício, além de receber novos exercícios com exemplos reais de análise de dados envolvendo as áreas de Data Science, Econometria, Machine Learning, Macroeconomia Aplicada, Finanças Quantitativas e Políticas Públicas diretamente em seu e-mail.

Compartilhe esse artigo

Comente o que achou desse artigo

Boletim AM

Encontre o seu conteúdo

Categorias

Artigos mais acessados

Regimes da Política Monetária Brasileira com Markov Switching no Python

Como criar janelas móveis de séries temporais usando o Python

Como incorporar choques em cenários de previsão?

Como preparar os dados para um modelo preditivo?

Transfer Learning para Previsão de Séries Temporais com o Python

Como usar o Python para tratar e manipular dados financeiros

Outros artigos relacionados

Regimes da Política Monetária Brasileira com Markov Switching no Python

Este exercício analisa a política monetária brasileira utilizando modelos de Markov Switching Regression. O objetivo é identificar diferentes regimes de política monetária e como eles influenciam a taxa Selic, a meta de inflação e o hiato do produto. Usamos a linguagem de programação Python para o processo de coleta, tratamento, análise e modelagem dos dados.

Como criar janelas móveis de séries temporais usando o Python

Janelas Móveis/Deslizantes, ou Rolling Windows, são termos frequentes na análise de séries temporais. Mas o que são e como aplicá-las no Python? Neste tutorial, mostramos como essa ferramenta é essencial para a análise de dados utilizando como exemplo a correlação móvel de ações brasileiras.

Como incorporar choques em cenários de previsão?

Neste exercício mostramos como incorar choques no cenário de variáveis exógenas para fins de previsão. Usando como exemplo a previsão do IPCA, através de um modelo de machine learning, mostramos os cuidados a serem tomados e uma forma simples de definir o cenário com os choques. Ao final, apresentamos uma previsão com um suposto choque e uma previsão sem o choque para comparação.

Boletim AM

Receba diretamente em seu e-mail gratuitamente nossas promoções especiais e conteúdos exclusivos sobre Análise de Dados!

Boletim AM

Receba diretamente em seu e-mail gratuitamente nossas promoções especiais e conteúdos exclusivos sobre Análise de Dados!