Skip to content

2023 07 13

木の直径

【アルゴリズム】入れ子リストの行和・列和を取る方法

A = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
sum_h = list(map(sum, A))  # [6, 15, 24]
sum_w = list(map(sum, zip(*A)))  # [12, 15, 18]