How to calculate the products of (jagged) uneven-sized matrix columns in Python

1 Answer

0 votes
from math import prod

def column_products(matrix):
    # Find the maximum row length
    max_cols = max(len(row) for row in matrix)

    products = []
    for col in range(max_cols):
        # Collect all existing elements in this column
        col_values = [row[col] for row in matrix if col < len(row)]
        products.append(prod(col_values))

    return products


def main():
    matrix = [
         [1, 2, 3],
         [4, 5],
         [6, 7, 8, 9]
    ]

    result = column_products(matrix)
    print("Column products:", result)


if __name__ == "__main__":
    main()


   
'''
run:
 
Column products: [24, 70, 24, 9]
   
'''

 



answered Feb 12 by avibootz
...