Python入门教程(四十)Python的NumPy数组创建

  • Post category:Python

下面介绍一下“Python入门教程(四十)Python的NumPy数组创建”的完整攻略。

一、NumPy数组

NumPy是Python的一个科学计算库,它提供了一个高性能的多维数组对象,以及对数组进行操作的工具。在使用NumPy时,最基本的是使用它的数组对象,即ndarray,可以把这个对象看成是一个多维数组,它的元素都是同一种数据类型。

二、数组创建

在使用NumPy的数组对象之前,我们需要了解如何创建一个数组对象。NumPy提供了几种创建数组的方法。

1. 直接创建数组

可以使用numpy.array()方法直接创建一个数组,但此方法需要手动输入元素值。

import numpy as np

a = np.array([1, 2, 3])
print(a)    # [1 2 3]

2. arange方法创建一维数组

NumPy提供了arange()方法创建一维等差数组,可以指定其起始位置、终止位置和步长。

import numpy as np

a = np.arange(0, 10, 2)
print(a)    # [0 2 4 6 8]

3. linspace方法创建一维数组

使用np.linspace()方法可以创建等差数列,可以指定起始位置、终止位置和元素个数。其中,元素个数指定后,这个区间就被分成了指定个数的小段。

import numpy as np

a = np.linspace(0, 1, 5)
print(a)    # [0.   0.25 0.5  0.75 1.  ]

4. zeros方法创建一维数组

可以使用np.zeros()方法创建元素全部为0的一维数组,但需要指定数组的长度。

import numpy as np

a = np.zeros((5,))
print(a)    # [0. 0. 0. 0. 0.]

5. ones方法创建一维数组

可以使用np.ones()方法创建元素全部为1的一维数组,但需要指定数组的长度。

import numpy as np

a = np.ones((5,))
print(a)    # [1. 1. 1. 1. 1.]

三、示例

示例1

使用NumPy创建一个二维数组arr,数组的第一维长度为3,第二维长度为4,元素全部为0。

import numpy as np

arr = np.zeros((3, 4))
print(arr)

输出结果为:

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

示例2

使用NumPy创建一个一维数组arr,数组的元素为0到9,步长为2。

import numpy as np

arr = np.arange(0, 10, 2)
print(arr)

输出结果为:

[0 2 4 6 8]

以上是“Python入门教程(四十)Python的NumPy数组创建”的完整攻略,希望能够帮助到你。