import pandas as pd def read_data(file_path): """ Read data from a csv file """ return pd.read_csv(file_path, sep=",") if __name__ == "__main__": file_path = "data/mmlu/response_rec.csv" data = read_data(file_path) high_school_physics = data[data['sub_topic'] == 'high_school_physics'] print(high_school_physics.head(5))