Find the probability mass function of a random variable X corresponding to the absolute value of the difference between the results of two dice. code example

Example: Write a function called square_odd that has one parameter. Your function must calculate the square of each odd number in a list. Return a Python 3 list containing the squared values Challenge yourself: Solve this problem with a list comprehension!

nums = [square_odds**2 for square_odds in nums if square_odds%2 != 0]