Load case/control/neither from csv column, fix Χ² filter.

[lightning.git] / chisquare.go

diff --git a/chisquare.go b/chisquare.go
index e1c8bfff1b6f5b84ab77b6b23970817064f0b94a..31142a1710686d2763f794cff54d0a27debcd3fa 100644 (file)
--- a/chisquare.go
+++ b/chisquare.go
@@ -32,6 +32,9 @@ func pvalue(a, b []bool) float64 {
                for bi := 0; bi < 2; bi++ {
                        rowtotal := tab[ai*2] + tab[ai*2+1]
                        coltotal := tab[bi] + tab[2+bi]
+                       if rowtotal == 0 || coltotal == 0 {
+                               return 1
+                       }
                        exp := float64(rowtotal) * float64(coltotal) / float64(len(a))
                        obs := tab[ai*2+bi]
                        d := float64(obs) - exp