some python assignments using numpy