上礼拜还在头疼Python原生不支持Matrix,写起来麻烦。这周看到了内嵌列表推导式,非常舒适。

比如写一个全零矩阵。

zeros = [ [0 for col in range(5)] for row in range(4)]

意味着这个0,作为一个表达式,作用于后面两次迭代。0不能理解为一个整数,而是看作一个表达式。

那写一个index相加的矩阵就变成了:

matrices = [ [r+v for v in range(matrix_width] for v in range(matrix_height)]

很方便,单个赋值就是分别索引,比如右上角的元素索引就是matrices[height-1][width-1]

 

 
posted on 2021-07-30 11:10  bathcor  阅读(325)  评论(0)    收藏  举报