Python中的numpy.ones()

时间:2020-02-23 14:42:21  来源:igfitidea点击:

Python numpy.ones()函数返回给定形状和数据类型的新数组,其中元素的值设置为1。
此函数与numpy zeros()函数非常相似。

numpy.ones()函数参数

numpy.ones()函数的语法为:

ones(shape, dtype=None, order='C')
  • 形状是一个int或者一个int元组,用于定义数组的大小。
    如果我们仅指定一个int变量,则将返回一维数组。
    对于整数元组,将返回给定形状的数组。

  • dtype是一个可选参数,默认值为float。
    用于指定数组的数据类型,例如int。

  • 该顺序定义是在内存中以行优先(C风格)还是列优先(Fortran风格)顺序存储多维数组。

Python numpy.ones()示例

让我们看一些使用numpyones()函数创建数组的示例。

1.用一维数组创建一维数组

import numpy as np

array_1d = np.ones(3)
print(array_1d)

输出:

[1. 1. 1.]

请注意,元素的默认数据类型为float。
这就是数组中1.的原因。

2.创建多维数组

import numpy as np

array_2d = np.ones((2, 3))
print(array_2d)

输出:

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

3.具有int数据类型的NumPy个数组

import numpy as np

array_2d_int = np.ones((2, 3), dtype=int)
print(array_2d_int)

输出:

[[1 1 1]
 [1 1 1]]

4.具有元组数据类型和一个的NumPy数组

我们可以将数组元素指定为元组,也可以指定其数据类型。

import numpy as np

array_mix_type = np.ones((2, 2), dtype=[('x', 'int'), ('y', 'float')])
print(array_mix_type)
print(array_mix_type.dtype)

输出:

[[(1, 1.) (1, 1.)]
 [(1, 1.) (1, 1.)]]
[('x', '<i8'), ('y', '<f8')]