Estou na fase inicial dos meus estudos sobre análise de dados e nesse módulo eu não consegui progredir. estou tendo muita dificuldade em extrair os dados como solicitado no exercício, se alguém puder me ajudar agradeço muito. Estou usando Google Colab.
%%writefile carros.csv
id,valor_venda,valor_manutencao,portas,pessoas,porta_malas
1,vhigh,med,2,2,small
2,med,vhigh,2,2,small
3,low,vhigh,2,2,small
4,low,high,2,2,small
5,low,high,2,2,small
6,low,high,4,4,big
7,low,high,4,4,big
8,low,med,2,2,small
9,low,med,2,2,small
10,low,med,2,2,small
11,low,med,4,4,big
12,low,low,2,2,small
13,low,low,4,4,small
14,low,low,4,4,med
Complete a função abaixo para extrair uma coluna do arquivo csv em uma lista. Os elementos devem ter o tipo de dado correto:
def extrai_coluna_csv(nome_arquivo: str, indice_coluna: int, tipo_dado: str):
coluna = []
# leia o arquivo com o comando 'with' utilizando o parametro 'nome_arquivo'
# extraia a coluna do arquivo utilizando o parametro 'indice_coluna'
# use a estrutura de decisão if/elif/else para fazer a conversão do tipo de dados utilizando o parametro 'tipo_dado'
return coluna
Você pode testar a função com os códigos abaixo:
# extrair a coluna valor_venda
valor_venda = extrai_coluna_csv(nome_arquivo='./carros.csv', indice_coluna=1, tipo_dado='str')
print(valor_venda) # deve retornar ['vhigh', 'med', 'low', ...]
# extrair a coluna pessoas
pessoas = extrai_coluna_csv(nome_arquivo='./carros.csv', indice_coluna=4, tipo_dado='int')
print(pessoas) # deve retornar [2, 2, 2, ...]