matplotlib_with_parquet

Peroxide + Matplotlib + Parquet

Prerequisites

• Python libraries
  • matplotlib
  • pandas
  • pyarrow

Process

Let's draw a multiple line plot - $y=x,~y=x^2,~y=x^3$.

File structure is as follows.

• Project root
  • src
    • main.rs
  • data (Empty directory - we will write data into this folder)
  • Cargo.toml

1. Main code (src/main.rs)

   use peroxide::fuga::*;
   
   // Let's draw a plot y = x, y = x^2, y=x^3
   
   fn main() -> Result<(), Box<dyn Error>> {
       // First make a domain vector (0 to 1 with stepsize 0.01)
       let x = seq(0, 1, 0.01);
       
       // Second make x^2
       let x2 = x.fmap(|t| t.powi(2));
   
       // Third make x^3
       let x3 = x.fmap(|t| t.powi(3));
   
       // Create dataframe
       let mut df = DataFrame::new(vec![]);
       df.push("x", Series::new(x));
       df.push("x2", Series::new(x2));
       df.push("x3", Series::new(x3));
   
       // Write dataframe to parquet format
       df.write_parquet("data/plot.parquet")?;
   
       Ok(())
   }

2. Run Main code

   cargo run --release

3. Python Code (pq_plot.py)

   import matplotlib.pyplot as plt
   import pandas as pd
   
   # Use latex
   plt.rc('text', usetex=True)
   plt.rc('font', family='serif')
   
   # Import parquet file
   df = pd.read_parquet('./data/plot.parquet')
   
   # Prepare Data to Plot
   x = df['x']
   x2 = df['x2']
   x3 = df['x3']
   
   # Prepare Plot
   plt.figure(figsize=(10,6), dpi=300)
   plt.title(r"Title", fontsize=16)
   plt.xlabel(r'$x$', fontsize=14)
   plt.ylabel(r'$y$', fontsize=14)
   
   # Plot with Legends
   plt.plot(x, x, label=r'$y=x$')
   plt.plot(x, x2, label=r'$y=x^2$')
   plt.plot(x, x3, label=r'$y=x^3$')
   
   # Other options
   plt.legend(fontsize=12)
   plt.grid()
   plt.savefig("plot.png", dpi=300)

4. Run python code

   python pq_plot.py

5. Finish - there is plot.png now.