下面介绍一下“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数组创建”的完整攻略,希望能够帮助到你。