欢迎访问宙启技术站
智能推送

如何使用Python函数创建一个矩阵?

发布时间:2023-06-17 09:49:25

在Python中,我们可以使用内置模块numpy(Numerical Python),使用Python函数创建一个矩阵是非常简单的,只需要使用numpy中的函数即可。在numpy中,矩阵表示为ndarray对象,可以使用以下函数创建矩阵:

1. numpy.array()函数

通过使用numpy.array函数,我们可以从现有的Python列表创建一个矩阵。这个函数使用列表作为输入参数,并将其转换为矩阵。下面是使用numpy.array创建矩阵的代码:

import numpy as np
a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print(a)

输出结果:

[[1 2 3]
 [4 5 6]
 [7 8 9]]

2. numpy.zeros()函数

numpy.zeros返回新的以0填充的数组。该函数可以使用一个整数或一个元组等参数创建一个矩阵。其中,参数是一个整数时,返回的矩阵是一个只含0的一维数组,当参数是一个元组的时候,返回一个以0填充的多维数组。

import numpy as np
a = np.zeros((3, 4)) #创建一个3行4列的矩阵
print(a)

输出结果:

[[0. 0. 0. 0.]
 [0. 0. 0. 0.]
 [0. 0. 0. 0.]]

3. numpy.ones()函数

与numpy.zeros函数类似,numpy.ones函数返回新的以1填充的数组。它使用一个整数或一个元组等参数来创建一个矩阵。

import numpy as np
a = np.ones((3, 4)) #创建一个3行4列的矩阵
print(a)

输出结果:

[[1. 1. 1. 1.]
 [1. 1. 1. 1.]
 [1. 1. 1. 1.]]

4. numpy.eye()函数

使用numpy.eye函数,我们可以创建一个单位矩阵。单位矩阵是一个n×n矩阵,其中对角线上的元素都是1,其余元素都是0,形象地表示为一个盒子中的“1”。

import numpy as np
a = np.eye(4) #创建一个4行4列的矩阵
print(a)

输出结果:

[[1. 0. 0. 0.]
 [0. 1. 0. 0.]
 [0. 0. 1. 0.]
 [0. 0. 0. 1.]]

5. numpy.arange()函数

numpy.arange函数用于从给定的范围返回均匀间隔的值。该函数使用start,stop和step作为参数。start表示起始值,stop表示终止值,step表示两个连续元素之间的步长。下面是使用numpy.arange函数创建矩阵的代码:

import numpy as np
a = np.arange(1, 13, 1).reshape(3, 4) #创建一个3行4列的矩阵
print(a)

输出结果:

[[ 1  2  3  4]
 [ 5  6  7  8]
 [ 9 10 11 12]]

6. numpy.random.rand()函数

numpy.random.rand函数生成给定大小的随机值。它可以使用整数或元组作为参数来创建矩阵。该函数返回一个均匀分布的随机样本。

import numpy as np
a = np.random.rand(3, 4) #创建一个3行4列的矩阵
print(a)

输出结果:

[[0.49405114 0.38746747 0.67797445 0.17616581]
 [0.88411697 0.81003485 0.89797846 0.75436152]
 [0.00629791 0.55183866 0.92850995 0.57331179]]

总结:

以上给出的六个函数,可以满足创建矩阵的绝大部分需求,其中numpy.array函数是最基本的创建矩阵方法。numpy.zeros,numpy.ones,numpy.eye是创建特殊矩阵时使用的函数,numpy.arange函数用于创建均匀分布的矩阵,在实际应用中比较常见。numpy.random.rand函数则用于创建随机矩阵。通过上述函数,我们可以方便地创建矩阵,为后续的数据分析提供了极大的便利。