How to change the code from Scala to Python

Question

def computeTotalVariationDistance(p: Distribution, q: Distribution): Double = {
    val pSum = p.sum
    val qSum = q.sum
    val l1Distance = p.zip(q)
      .map { case (_, pVal, qVal) =>
        math.abs((pVal / pSum) - (qVal / qSum))
      }
      .sum

    0.5 * l1Distance
  }

Can someone help me to change this code into python.

Answer 1

It's actually relatively straightforward. Instead of map, you can use list comprehension:

l1Distance = sum(
             [abs((pVal / pSum) - (qVal / qSum)) 
              for pVal, qVal in zip(p, q)]
             )

I have not tested this but it should work, or something very similar.