heatmap: handles pandas and numpy masks