Correlation heatmap

by

Another alternative is to use the heatmap function in seaborn to plot the covariance. This example uses the Auto data set from the ISLR package in R (the same as in the example you showed).

import pandas.rpy.common as com
import seaborn as sns
%matplotlib inline

# load the R package ISLR
infert = com.importr("ISLR")

# load the Auto dataset
auto_df = com.load_data('Auto')

# calculate the correlation matrix
corr = auto_df.corr()

# plot the heatmap
sns.heatmap(corr, 
        xticklabels=corr.columns,
        yticklabels=corr.columns)

enter image description here

If you wanted to be even more fancy, you can use Pandas Style, for example:

cmap = cmap=sns.diverging_palette(5, 250, as_cmap=True)

def magnify():
    return [dict(selector="th",
                 props=[("font-size", "7pt")]),
            dict(selector="td",
                 props=[('padding', "0em 0em")]),
            dict(selector="th:hover",
                 props=[("font-size", "12pt")]),
            dict(selector="tr:hover td:hover",
                 props=[('max-width', '200px'),
                        ('font-size', '12pt')])
]

corr.style.background_gradient(cmap, axis=1)\
    .set_properties(**{'max-width': '80px', 'font-size': '10pt'})\
    .set_caption("Hover to magify")\
    .set_precision(2)\
    .set_table_styles(magnify())

enter image description here

More Related Contents:

  1. Computing the correlation coefficient between two multi-dimensional arrays
  2. LabelEncoder for categorical features?
  3. How to calculate correlation between all columns and remove highly correlated ones using pandas?
  4. Use .corr to get the correlation between two columns
  5. Pandas Correlation Groupby
  6. pandas columns correlation with statistical significance
  7. find time shift between two similar waveforms
  8. Why is “1000000000000000 in range(1000000000000001)” so fast in Python 3?
  9. How to have one colorbar for all subplots
  10. unknown error: session deleted because of page crash from unknown error: cannot determine loading status from tab crashed with ChromeDriver Selenium
  11. How do I specify new lines on Python, when writing on files?
  12. How can I iterate through two lists in parallel?
  13. How to resolve TypeError: can only concatenate str (not “int”) to str [duplicate]
  14. First non-null value per row from a list of Pandas columns
  15. Selecting multiple slices from a numpy array at once
  16. How do I change the range of the x-axis with datetimes in matplotlib?
  17. Private members in Python
  18. Disable assertions in Python
  19. When do you use ‘self’ in Python?
  20. APT command line interface-like yes/no input?
  21. Cleanest way to hide every nth tick label in matplotlib colorbar?
  22. Segmentation fault: 11 in OS X
  23. Python check for integer input
  24. How to delete a record by id in Flask-SQLAlchemy
  25. Getting all visible text from a webpage using Selenium
  26. Python & Selenium: Difference between driver.implicitly_wait() and time.sleep()
  27. Environment Variables when script run by cron
  28. Why is `with open()` better for opening files in Python?
  29. Matplotlib savefig with a legend outside the plot
  30. numpy faster than numba and cython , how to improve numba code

Leave a Comment