Leia CSV para Array em Python

O uso de arquivos CSV é amplamente difundido no campo de análise de dados / ciência de dados em Python. CSV significa Comma Separated Values. Esses tipos de arquivos são usados ​​para armazenar dados na forma de tabelas e registros. Nessas tabelas, existem várias colunas separadas por vírgulas. Uma das tarefas de manipulação desses arquivos CSV é importá-los na forma de arrayes de dados.

Este tutorial apresentará diferentes métodos para importar arquivos CSV na forma de arrayes de dados.

Use numpy.loadtxt() para ler um arquivo CSV em um array em Python

Como o nome sugere, a função open() é usada para abrir o arquivo CSV. A função loadtxt() do NumPy ajuda a carregar os dados de um arquivo de texto. Nos argumentos desta função, existem dois parâmetros que devem ser mencionados: o nome do arquivo ou a variável em que o nome do arquivo está armazenado, e o outro é denominado delimiter, que denota a string usada para separar os valores. O valor padrão do delimitador é o espaço em branco.

Exemplo:

import numpy as np

with open("randomfile.csv") as file_name:
    array = np.loadtxt(file_name, delimiter=",")

print(array)

Aqui, observe que o valor do delimitador foi definido como uma vírgula. Portanto, o separador na matriz retornada é uma vírgula.

Use o método list() para ler um arquivo CSV em um array 1D em Python

Aqui usamos o módulo csv do Python, que é usado para ler esse arquivo CSV no mesmo formato tabular. Mais precisamente, o método reader() deste módulo é usado para ler o arquivo CSV.

Finalmente, o método list() pega todas as sequências e os valores em formato tabular e os converte em uma lista.

Exemplo:

import csv

with open("randomfile.csv") as file_name:
    file_read = csv.reader(file_name)

array = list(file_read)

print(array)

Aqui, armazenamos os dados lidos pela função reader() em uma variável e usamos essa variável para converter esses dados em uma lista.