CSV to Array

[fahadqureshi]

i have a csv file that has a number in one column and a corresponding number in the second column. i am trying to store both of these numbers in an array. i have done this before with an excel file but how would i go about doing this with a csv file since the command xlsht wont work in csv. here is my code so far to print everything but how do i store them in an array and what would be the correct way to define the array.

import csv
reader = csv.reader(open("C:\Fahad\New\Dtap.csv", "rb"))
for row in reader:
    print row
 

[fahadqureshi]

ok i did figure out how to store it in an array but how do i store it as a number. it looks like python is just treating it like a character instead of a number.
also i want the array size to be equal to the number of columns in the csv file what would be the correct way to define that?

[bartonc]

ok i did figure out how to store it in an array but how do i store it as a number. it looks like python is just treating it like a character instead of a number.
also i want the array size to be equal to the number of columns in the csv file what would be the correct way to define that?

Here's a trick that I use. I hope you find it useful:

>>> data = ['12.2', 'hello', '7']
>>> converters = (float, str, int)
>>> for i, item in enumerate(data):
...     print converters[i](item)
...     
12.2
hello
7
>>> 

[bvdet]

This function is useful when you do not know the types of the data items:

def convertType(s):
        for func in (int, float, eval):
            try:
                n = func(s)
                return n
            except:
                pass
        return s

>>> data = ['12.2', 'hello', '7', '[12,24,36,"goodbye"]', '12*12']
>>> for item in data:
...     print convertType(item)
...     
12.2
hello
7
[12, 24, 36, 'goodbye']
144
>>> 

[bartonc]

This function is useful when you do not know the types of the data items:

def convertType(s):
        for func in (int, float, eval):
            try:
                n = func(s)
                return n
            except:
                pass
        return s

>>> data = ['12.2', 'hello', '7', '[12,24,36,"goodbye"]', '12*12']
>>> for item in data:
...     print convertType(item)
...     
12.2
hello
7
[12, 24, 36, 'goodbye']
144
>>> 

Nice touch, throwing in the eval! I really like it!

[bvdet]

Nice touch, throwing in the eval! I really like it!

I actually use this in almost every script run in SDS/2. We import data from a text file formatted like this:

#Python Script Default File
access_hole_height=1.25
access_hole_length=1.25
access_hole_radius=0.5
flange_setback_bott=6.5
flange_setback_top=6.5
flush_lower_left=0.0
flush_lower_right=0.0
flush_upper_left=0.5
flush_upper_right=0.5
prep_angle=45.0
prep_list=['Top Left', 'Bottom Left']
print_doc=No

Here's the actual code used:

def fdim(s):
    ss = s.split('/')
    return float(ss[0])/float(ss[1])
 
def convertType(s):
        for func in (int, float, fdim, eval):
            try:
                n = func(s)
                return n
            except:
                pass
        return s
 
def import_data(file_name):
    patt = r'.+=.+'
    try:
        data = [item.strip().split('=') for item in re.findall(patt, open(file_name).read()) if not item.startswith('#')]
        return dict(zip([j[0].strip() for j in data], [convertType(i[1].strip()) for i in data]))
    except:
        Warning("User default file import error. Reverting to auto or initial defaults...")
        return None

It returns a dictionary with the values of the correct types. The 'eval()' was necessary to convert saved lists, and the 'fdim()' to convert fractions that were saved to disk as strings.