Learn practical skills, build real-world projects, and advance your career
Created 4 years ago
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]