Python pandas import

Question

I have this import and then set the columns I want. I can in the import itself set the columns I want?

import pandas as pd 

data=pd.read_csv("C:/Users/t714591/Videos/comissoes/95/96.txt",encoding='ISO-8859-1', sep=';')

data = data.iloc[:, [0, 1, 4, 5, 6]]

data.to_excel(r'C:/Users/t714591/Videos/comissoes/95/teste.xlsx', index = False)

Answer 1

Yes, you can use the parameter usecols

Your code will look like this:

data=pd.read_csv("C:/Users/t714591/Videos/comissoes/95/96.txt",encoding='ISO-8859-1', sep=';', usecols=[0, 1, 4, 5, 6])

Information can be found directly in the documentation

https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.read_csv.html