Python中的 numpy.ones()

numpy.ones()是Python中numpy库中的一个函数，用于创建指定尺寸的全1数组或全1矩阵。它有许多参数可以控制数组创建的形状、数据类型和其他属性。

语法

numpy.ones(shape, dtype=None, order='C')

• shape：整数或整数元组，用于定义输出数组的形状
• dtype：可选的，用于指定输出数组的数据类型，默认为numpy.float64
• order：可选的，用于指定数组元素在存储器中的顺序，可为'C'（按行）或'F'（按列），默认为'C'

示例

基本用法

import numpy as np

# 一维数组
a = np.ones(5)
print(a)
# 输出
# array([1., 1., 1., 1., 1.])

# 二维数组
b = np.ones((3, 2))
print(b)
# 输出
# array([[1., 1.],
#        [1., 1.],
#        [1., 1.]])

# 三维数组
c = np.ones((2, 3, 2))
print(c)
# 输出
# array([[[1., 1.],
#         [1., 1.],
#         [1., 1.]],

#        [[1., 1.],
#         [1., 1.],
#         [1., 1.]]])

指定数据类型

import numpy as np

# 指定数据类型为int
a = np.ones(5, dtype=int)
print(a)
# 输出
# array([1, 1, 1, 1, 1])

# 指定数据类型为bool
b = np.ones((3, 2), dtype=bool)
print(b)
# 输出
# array([[ True,  True],
#        [ True,  True],
#        [ True,  True]])

指定存储顺序

import numpy as np

# 按列存储
a = np.ones((3, 2), order='F')
print(a)
# 输出
# array([[1., 1.],
#        [1., 1.],
#        [1., 1.]], order='F')

# 按行存储，与默认情况一致
b = np.ones((3, 2), order='C')
print(b)
# 输出
# array([[1., 1.],
#        [1., 1.],
#        [1., 1.]])

总结

numpy.ones()函数可以方便地创建指定尺寸的全1数组或全1矩阵，可以通过指定数据类型和存储顺序来满足不同的需求。