import h_transport_materials as htm
from h_transport_materials.plotting import plot
import matplotlib.pyplot as plt

# filter only tungsten and H
diffusivities = htm.diffusivities.filter(material=[htm.TUNGSTEN, htm.COPPER])

mat_to_colour = {
   htm.TUNGSTEN: "tab:grey",
   htm.COPPER: "tab:orange"
}

plot(diffusivities, colour_by="material", key_to_colour=mat_to_colour)

plt.yscale("log")
plt.show()