def choose_one_variable(kind_of_data_in_table, stations): if len(stations) > 1: for station in stations: print '-You have data for station %s' % (str(station)) print '' station_ID = question.digit_station('Enter the station ID that you want to use: ', stations) else: station_ID = stations[0] variables1 = [] stationIDs1 = [] filenames1 = [] kinds1 = [] variable1_specs = [] #'select variable from station' variable1 = select_variable(kind_of_data_in_table, station_ID) #'create variable specifications for variable' for i in kind_of_data_in_table: if station_ID in i[0]: kind = kind_of_variable(i, station_ID, variable1) variables1.append(variable1) stationIDs1.append(station_ID) filenames1.append(i[0]) kinds1.append(kind) variable1_specs = zip(variables1, filenames1, stationIDs1, kinds1) return variable1_specs
def choose_variables_for_correlation(kind_of_data_in_table, stations): variable1 = [] variable2 = [] variable1_specs = [] variable2_specs = [] variables1 = [] stationIDs1 = [] filenames1 = [] kinds1 = [] variables2 = [] stationIDs2 = [] filenames2 = [] kinds2 = [] kind = '' if len(stations) == 1: if not variable1: #'select var1 from station 1' variable1 = select_variable(kind_of_data_in_table, stations[0]) #'create variable specifications for var1 from station 1' for i in kind_of_data_in_table: kind = kind_of_variable(i, stations[0], variable1) variables1.append(variable1) stationIDs1.append(stations[0]) filenames1.append(i[0]) kinds1.append(kind) variable1_specs = zip(variables1, filenames1, stationIDs1, kinds1) if variable1: #'select var2 from station 1' variable2 = select_variable(kind_of_data_in_table, stations[0]) #'create variable specifications for var2 from station 1' for i in kind_of_data_in_table: kind = kind_of_variable(i, stations[0], variable2) variables2.append(variable2) stationIDs2.append(stations[0]) filenames2.append(i[0]) kinds2.append(kind) variable2_specs = zip(variables2, filenames2, stationIDs2, kinds2) if len(stations) == 2: if not variable1: #'select var1 from station 1' variable1 = select_variable(kind_of_data_in_table, stations[0]) #'create variable specifications for var1 from station 1' for i in kind_of_data_in_table: if str(stations[0]) in i[0]: kind = kind_of_variable(i, stations[0], variable1) variables1.append(variable1) stationIDs1.append(stations[0]) filenames1.append(i[0]) kinds1.append(kind) variable1_specs = zip(variables1, filenames1, stationIDs1, kinds1) if variable1: #'select var2 from station 2' variable2 = select_variable(kind_of_data_in_table, stations[1]) for i in kind_of_data_in_table: if str(stations[1]) in i[0]: kind = kind_of_variable(i, stations[1], variable2) variables2.append(variable2) stationIDs2.append(stations[1]) filenames2.append(i[0]) kinds2.append(kind) variable2_specs = zip(variables2, filenames2, stationIDs2, kinds2) return variable1_specs, variable2_specs