How To Find Shape Of Dataset In Python