Your friend needs to create a correlation plot with different characteristics of popular movies. Here's what they want to get:
Here's the code they've written:
plt.imshow(movies.corr())
plt.xticks(range(len(movies.corr())), movies.corr().columns)
plt.yticks(range(len(movies.corr())), movies.corr().columns)
plt.title('Movies \n')
However, the resulting plot isn't quite the same as the one above. What should they add to the code to get the heatmap they want?