初始化二维列表

初始化一个给定宽度和高度的二维列表,并设置初始值。

  • 使用列表推导式和 range() 生成 h 行,每行都是一个长度为 w 的列表,初始值为 val
  • 如果省略最后一个参数 val,默认值将设置为 None
def initialize_2d_list(w, h, val = None):
  return [[val for x in range(w)] for y in range(h)]

initialize_2d_list(2, 2, 0) # [[0, 0], [0, 0]]