exporting data into a column of an excel file in python

Question

here is my code:

import random
from random import shuffle

fileName = 'test'
dataFile = open("DD" + fileName+'.csv', 'w')
dataFile.write('trial,numbers') #data headers for .csv file

a = [1,2]

for i in range(4):
     shuffle(a)
     if a == [1,2]:
         num = 'top'
     else:
         num = 'bottom'

    print a
    print num

    info = []
    i = str(i)
    info.append(',')
    info.append(i)
    info.append(',')
    info.append(num)

    dataFile.writelines(info)

This code randomizes an order then labels it either 'top' or 'bottom' 4 times, and exports the a excel file. The problem is that when it exports it puts it in a row. What I would like to do is have it exported into a column so that they are under the headlines 'trial' and 'numbers'. Any help would be greatly appreciated. :)

Burhan Khalid · Accepted Answer · 2015-03-01 05:31:08Z

1

You are not writing any end-of-line characters, so everything will be in one line (which is why you are seeing it all as a row).

Using the csv module, you can simplify your code:

import csv
import random

a = [1,2]

with open('DD{}filename.csv'.format(fileName), 'w') as f:
    writer = csv.writer(f, delimiter=',')
    writer.writerow(['trial','numbers']) # Note this is a two element
                                         # list, not a string

    for i in range(4):
        random.shuffle(a)
        if a == [1,2]:
           num = 'top'
        else:
           num = 'bottom'

        writer.writerow([i, num])

answered Mar 1, 2015 at 5:31

Burhan Khalid

175k20 gold badges254 silver badges291 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

exporting data into a column of an excel file in python

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related