Netflix! What started in 1997 as a DVD rental service has since exploded into one of the largest entertainment and media companies.
Given the large number of movies and series available on the platform, it is a perfect opportunity to flex your exploratory data analysis skills and dive into the entertainment industry. Our friend has also been brushing up on their Python skills and has taken a first crack at a CSV file containing Netflix data. They believe that the average duration of movies has been declining. Using your friends initial research, you'll delve into the Netflix data to see if you can determine whether movie lengths are actually getting shorter and explain some of the contributing factors, if any.
You have been supplied with the dataset netflix_data.csv , along with the following table detailing the column names and descriptions. This data does contain null values and some outliers, but handling these is out of scope for the project. Feel free to experiment after submitting!
The data
netflix_data.csv
| Column | Description |
|---|---|
show_id | The ID of the show |
type | Type of show |
title | Title of the show |
director | Director of the show |
cast | Cast of the show |
country | Country of origin |
date_added | Date added to Netflix |
release_year | Year of Netflix release |
duration | Duration of the show in minutes |
description | Description of the show |
genre | Show genre |
# Importing pandas and matplotlib
import pandas as pd
import matplotlib.pyplot as plt
# Start coding!
# Load the data from 'netflix_data.csv' into netflix_df
netflix_df = pd.read_csv('netflix_data.csv', index_col = 0)
print(netflix_df.head(5)) # print the first ten rows of the dataset# Filter out TV shows from the netflix_df
netflix_subset = netflix_df[netflix_df['type'] == 'Movie']
# Keeping only the specified columns for movies
netflix_movies = netflix_subset[['title', 'country', 'genre', 'release_year', 'duration']]
# Filter to find movies shorter than 60 minutes
short_movies = netflix_movies[netflix_movies['duration'] < 60]
# Initialize an empty list to store colors
colors = []
# Iterate through netflix_movies DataFrame to assign colors based on genre
for genre in netflix_movies['genre']:
if genre == 'Children':
colors.append('blue')
elif genre == 'Documentaries':
colors.append('green')
elif genre == 'Stand-Up':
colors.append('red')
else:
colors.append('gray')
# Initialize a matplotlib figure object
fig, ax = plt.subplots()
# Create a scatter plot for movie duration by release year
ax.scatter(netflix_movies['release_year'], netflix_movies['duration'], c=colors)
# Setting the labels and title for the plot
ax.set_xlabel('Release year')
ax.set_ylabel('Duration (min)')
ax.set_title('Movie Duration by Year of Release')
# Display the plot
plt.show()
# After inspecting the plot, assign either "yes" or "no" to the variable answer
# regarding the question "Are we certain that movies are getting shorter?"
answer = "no"