Python数据分析从入门到精通(第2版)
上QQ阅读APP看书,第一时间看更新

3.2.1 创建简单的数组

【例3.1】演示如何创建数组(实例位置:资源包\TM\sl\03\01)

NumPy创建简单数组主要使用array()函数,效果如图3.6所示。

图3.6 简单数组

程序代码如下:

     1 import numpy as np              # 导入numpy 模块
     2 n1 = np.array([1,2,3])          # 创建一个简单的一维数组
     3 n2 = np.array([0.1,0.2,0.3])    # 创建一个包含小数的一维数组
     4 n3 = np.array([[1,2],[3,4]])    # 创建一个简单的二维数组
1.为数组指定数据类型

【例3.2】为数组指定数据类型(实例位置:资源包\TM\sl\03\02)

NumPy支持比Python更多种类的数据类型,通过dtype参数可以指定数组的数据类型,程序代码如下:

     1   import numpy as np                            # 导入numpy 模块
     2   list = [1, 2, 3]                              # 列表
     3   n1 = np.array(list,dtype=np.float_)           # 创建浮点型数组
     4   # 或者
     5   n1= np.array(list,dtype=float)
     6   print(n1)
     7   print(n1.dtype)
     8   print(type(n1[0]))

运行程序,输出结果为:

     [1. 2. 3.]
     float64
     <class 'numpy.float64'>
2.数组的复制

【例3.3】复制数组(实例位置:资源包\TM\sl\03\03)

当运算和处理数组时,为了不影响原数组,就需要对原数组进行复制,而对复制后的数组进行修改、删除等操作都不会影响原数组。数组的复制可以通过copy参数来实现,程序代码如下:

运行程序,输出结果为:

     [1 2 3]
     [3 2 1]

数组n2是数组n1的副本,从运行结果得知:虽然修改了数组n2,但是数组n1没有发生变化。

3.通过ndmin参数控制最小维数

无论给出的数据维数是多少,ndmin参数都会根据最小维数创建指定数组。

【例3.4】修改数组的维数(实例位置:资源包\TM\sl\03\04)

假设ndmin=3,则即便给出的数组是一维的,仍会创建一个三维数组。程序代码如下:

运行程序,输出结果为:

     [[[1 2 3]]]