Row selection by input values in pandas

Viewed 98

I'm getting input from user and trying to select the specific row in a data frame. But I cant.

My code:

batteryserialnumber_in = input("Enter Batter_id [INSMOMAHXXXXXXXXXXXX]: ")
Time_in  = input("Enter Time [Hr:Min:Sec]: ")
Date_in  = input("Enter the date YYY-MM-DD format:")

res2 = df1.isin(df1['Date']=='Date_in' & df1['batteryserialnumber']== batteryserialnumber_in') &  df1['Time']=='Time_in'

My data frame has the date [YYYY-MM-DD] and battery serial, but no time [hr:mm:sec] column.

1 Answers

If you want to return only the rows in the dataframe that satisfies your filter condition based on the user input the following should do the trick:

batteryserialnumber_in = input("Enter Batter_id [INSMOMAHXXXXXXXXXXXX]: ")
Time_in = input("Enter Time [Hr:Min:Sec]: ") 
Date_in = input("Enter the date YYY-MM-DD format:")

res2 = df1.loc[(df1['Date']==Date_in) & (df1['batteryserialnumber']==batteryserialnumber_in) & (df1['Time']==Time_in)]
Related