implemented data analysis in pandas, more efficient?
test