python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > python生成二维矩阵

python生成二维矩阵的两种方法小结

作者:wu_zhiyuan

本文主要介绍了python生成二维矩阵,包含列表生成m行n列的矩阵和numpy生成想要维度的矩阵的两种方法,具有一定的参考价值,感兴趣的可以了解一下

1、直接用列表生成m行n列的矩阵

m,n = map(int,input().split())
matrix = [[0]*(m)]*(n)

输出为:

在这里插入图片描述

注意:这种方式生成的矩阵存在一定的问题,比如,无法给特定位置的元素赋值,例如:

matrix[1][1] = 9

输出为:

在这里插入图片描述

可见,第二列的元素全部被赋值为9了

2、采用numpy生成想要维度的矩阵

import numpy as np

x,y = map(int,input().split())
a = np.ones((x+1,y+1))

输出为:

在这里插入图片描述

上面的输出特别漂亮,一个完美的矩阵形式输出,下面我们试一下修改特定位置的元素值。

import numpy as np

x,y = map(int,input().split())
a = np.ones((x+1,y+1))  # 参数为元组 (x+1,y+1)
a[1][1] = 9
print(a)

输出为:

在这里插入图片描述

可见,我们成功修改了第二行第二列的元素值,因此通过numpy生成的矩阵更具有可操作性。比如下面的操作:

import numpy as np

x,y = map(int,input().split())
a = np.ones((x+1,y+1))
for i in range(1,x+1):
    for j in range(1,y+1):
        a[i][j] = a[i-1][j] + a[i][j-1]
print(a)

输出为:

在这里插入图片描述

到此这篇关于python生成二维矩阵的方法小结的文章就介绍到这了,更多相关python生成二维矩阵内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家! 

您可能感兴趣的文章:
阅读全文