CODEWITHSUNDEEP
pythoncsv
Sakil Chowdhury
Sakil Chowdhury

Reputation: 327

Append a Header for CSV file?

I am trying to add a header to my CSV file.

I am importing data from a .csv file which has two columns of data, each containing float numbers. Example:

  11   22
  33   44
  55   66

Now I want to add a header for both columns like: 

 ColA  ColB
  11    22
  33    44
  55    66

I have tried this: 

with open('mycsvfile.csv', 'a') as f:
    writer = csv.writer(f)
    writer.writerow(('ColA', 'ColB'))

I used 'a' to append the data, but this added the values in the bottom row of the file instead of the first row. Is there any way I can fix it?

Upvotes: 16

Views: 86469

Answers (6)

Shruti Sharma
Shruti Sharma

Reputation: 45

For the issue where the first row of the CSV file gets replaced by the header, we need to add an option.

import pandas as pd  
df = pd.read_csv('file.csv', **header=None**)  
df.to_csv('file.csv', header = ['col1', 'col2'])

Upvotes: 2

Mishal Ahmed
Mishal Ahmed

Reputation: 191

I know the question was asked a long time back. But for others stumbling across this question, here's an alternative to Python.

If you have access to sed (you do if you are working on Linux or Mac; you can also download Ubuntu Bash on Windows 10 and sed will come with it), you can use this one-liner:

sed -i 1i"ColA,ColB" mycsvfile.csv

The -i will ensure that sed will edit in-place, which means sed will overwrite the file with the header at the top. This is risky.

If you want to create a new file instead, do this

sed 1i"ColA,ColB" mycsvfile.csv > newcsvfile.csv

Upvotes: 8

Ketan Goyani
Ketan Goyani

Reputation: 1

You can set reader.fieldnames in your code as list like in your case 

 with open('mycsvfile.csv', 'a') as fd:
        reader = csv.DictReader(fd)
        reader.fieldnames = ["ColA" , "ColB"]
        for row in fd

Upvotes: -1

Mark Tolonen
Mark Tolonen

Reputation: 178409

One way is to read all the data in, then overwrite the file with the header and write the data out again. This might not be practical with a large CSV file:

#!python3
import csv
with open('file.csv',newline='') as f:
    r = csv.reader(f)
    data = [line for line in r]
with open('file.csv','w',newline='') as f:
    w = csv.writer(f)
    w.writerow(['ColA','ColB'])
    w.writerows(data)

Upvotes: 33

Hai Vu
Hai Vu

Reputation: 40783

In this case, You don't need the CSV module. You need the fileinput module as it allows in-place editing:

import fileinput

for line in fileinput.input(files=['mycsvfile.csv'], inplace=True):
    if fileinput.isfirstline():
        print 'ColA,ColB'
    print line,

In the above code, the print statement will print to the file because of the inplace=True parameter.

Upvotes: 2

adrianX
adrianX

Reputation: 627

i think you should use pandas to read the csv file, insert the column headers/labels, and emit out the new csv file. assuming your csv file is comma-delimited. something like this should work:

   from pandas import read_csv

   df = read_csv('test.csv')
   df.columns = ['a', 'b']
   df.to_csv('test_2.csv')

Upvotes: 10

Related Questions