Python iterating over dataframe based on user input

Question

Im trying to filter the data frame (stops_trips_vehicles) based on user input - specifying hour (departure_time column) and the stop name(stop_name column). Each time I am getting wrong value as a result, even if the input data is matching in the data frame.

hour = input('Input hour ') 
stop = input('Input stop name ')

results = [] 
for index, row in stops_trips_vehicles.iterrows():
    if row['stop_name'] == stop and row['departure_time'] == hour:
        results.append(row)
        print(results)
    else:
        print('wrong value')
        break

speeder1987 · Accepted Answer · 2022-05-13 11:20:36Z

1

You don't need to filter the DataFrame row by row:

hour = input('Input hour ') 
stop = input('Input stop name ')

results = stops_trips_vehicles[(stops_trips_vehicles['stop_name']==stop) & (stops_trips_vehicles['departure_time']==hour)]

answered May 13, 2022 at 11:20

speeder1987

3173 silver badges13 bronze badges

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

Collectives™ on Stack Overflow

Python iterating over dataframe based on user input

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