Python beginner, Data analysis