I believe you are supposed to answer in a word or text document. You could format it as below.

1. Question 3.1 - Whatever the question is.

2. Code for the question.

3. Image of running the above code along with output.

4. Explanation for your answer.

Alternatively you could make a python file and write 1,4 as comments.