Python | sympy.ones() 方法

sympy.ones()方法返回一个具有指定形状和数据类型的数组，该数组的所有元素都设置为1。此方法可用于创建全为1的矩阵。

语法

sympy.ones(shape, dtype=int)

参数

• shape：指定数组的形状。
• dtype：（可选参数）指定数组的元素类型。默认为int。

返回值

返回一个sympy.ndarray类型的数组，其中所有元素都为1。

示例

import sympy

# 创建一个3行3列的全1矩阵
matrix1 = sympy.ones(3, 3)
print(matrix1)

# 创建一个2行3列的全1矩阵，元素类型为float
matrix2 = sympy.ones((2, 3), dtype=float)
print(matrix2)

输出结果：

Matrix([[1, 1, 1], [1, 1, 1], [1, 1, 1]])
Matrix([
[1.0, 1.0, 1.0],
[1.0, 1.0, 1.0]])

以上示例中，sympy.ones(3, 3)创建一个3行3列的全1矩阵，并将其赋值给变量matrix1；sympy.ones((2, 3), dtype=float)创建一个2行3列的全1矩阵，并将其赋值给变量matrix2，其中元素类型为float。

此外，sympy.ones()也可以用于创建一维数组。例如，sympy.ones(5)将创建一个长度为5的一维数组，其中所有元素都为1。