Python生成10x10随机数矩阵如何优化写法,减少重复代码行数
10×10随机矩阵代码优化方案
方案1:纯Python标准库实现(无第三方依赖)
你可以直接使用嵌套列表推导替代重复的10行列表定义,功能完全一致,代码更精简:
import random a = [[random.randint(1, 1000) for _ in range(10)] for _ in range(10)] # 输出也可简化为如下写法 for row in a: print(" ".join(map(str, row)))
说明
- 外层的
for _ in range(10)控制生成10行,内层列表推导每行生成10个符合范围的随机数 - 用
_作为循环变量是Python惯例,代表该变量不会被后续逻辑使用 - 输出部分用字符串join方法拼接行元素,比双层循环写法更简洁高效
方案2:基于numpy实现(适合有数值计算需求的场景)
如果你的程序后续需要对矩阵做数值运算,可以使用numpy库实现,代码更短:
import numpy as np a = np.random.randint(1, 1001, size=(10, 10)) print(a)
说明
- numpy的
randint方法的high参数为开区间,因此要生成最大为1000的数需要设置high=1001 - size参数直接指定矩阵维度,不需要手动写循环逻辑,输出也自带矩阵格式化效果
内容的提问来源于stack exchange,提问作者John35
相关产品推荐
相关产品推荐

