Get size of vector numpy