Basics of visualization in Python