CSV, lists, and functions
C.T.
swilks06 at gmail.com
Tue Mar 19 18:59:26 EDT 2013
Hello,
Currently doing a project for class an I'm stuck. I have a csv file that I'm suppose to extract some information from. I've created a function that ignores the first six lines of the csv file and creates a list of values in a particular column. Here is the code:
def get_values(file, index):
'''(file object, int) -> list
Return a list of states and corresponding values at a prticular index in file.'''
values_list = []
file.readlines(900)#skipping first 6 lines of file
for line in file:
line_list = line.split(',')
values_list.append(line_list[index])
return values_list
#Prompt for file and prompt again if file is not found
while True:
try:
file_name = input('Enter in file name: ')
input_file = open( file_name, 'r')
break
except IOError:
print('File not found.')
heart_list = get_values(input_file, 1)
motor_list = get_values(input_file, 4)
I can see a list of values for heart_list, but when I print the motor_list I get back an empty list. I think its because I'm not closing the file before calling the function again, but I don't want to prompt for the same file over and over again because I need to pull at least 10 columns from this csv files and turn them into lists. Any help is appreciated.
More information about the Python-list
mailing list