I’ve posted notes on viability selection at one locus with two alleles: http://darwin.eeb.uconn.edu/eeb348-notes/selection.pdf. I also wrote and posted a simply R Shiny application that allows you to see the within-generation effect of viability selection using a fixed set of genotypes and sliders to pick viabilities: https://keholsinger.shinyapps.io/Viability-selection/. Enjoy!