plotting heatmaps in Haskell

{-# LANGUAGE QuasiQuotes #-}
{-# LANGUAGE LambdaCase #-}
 
module Plotting (plotHeatMap) where
 
import Data.List
import Data.Text qualified as TText
import Data.String.Interpolate
import Data.String.Here
import qualified Data.Set.Monad as Set
import System.Process
import Text.RawString.QQ
import Control.Monad
 
fromPython :: String -> IO String
fromPython code = readProcess "python3" ["-c", code] ""
 
executePython :: String -> IO ()
executePython code = do
  _ <- readProcess "python3" ["-c", code] ""
  return ()
 
example :: [[Double]]
example = 
    [[1, 2, 3, 4, 4],
     [4, 2, 3, 3, 4],
     [9, 1, 0, 10, 2]]  
 
plotHeatMap :: [[Double]] -> IO ()
plotHeatMap map = do
  executePython [__i|
    import numpy as np
    import seaborn as sns
    import matplotlib.pyplot as plt
    sns.set_theme(style="white")
    cmap = sns.diverging_palette(230, 20, as_cmap=True)
    heatmap_data = np.array(#{map})
    sns.heatmap(heatmap_data, cmap=cmap)
    plt.show()|]

Works via the matplotlib Python library, it actually calls the seaborn Python library.

Tags: Haskell,