Learn practical skills, build real-world projects, and advance your career
import pandas as pd
test_data = pd.read_csv(r'..\data\drugsComTest_raw.csv')
test_data.info()
<class 'pandas.core.frame.DataFrame'> RangeIndex: 53766 entries, 0 to 53765 Data columns (total 7 columns): uniqueID 53766 non-null int64 drugName 53766 non-null object condition 53471 non-null object review 53766 non-null object rating 53766 non-null int64 date 53766 non-null object usefulCount 53766 non-null int64 dtypes: int64(3), object(4) memory usage: 2.9+ MB
cond1= test_data['drugName'] == 'Bupropion'
cond2= test_data['drugName'] == 'Sertraline'
f_testdf = test_data[cond1 | cond2]