Numpy(4)

多维数组

'''
# coding=utf-8  

Created on 2016-9-6 

@author: paulsweet 

'''
import numpy as np

#生成6*6的矩阵
a=np.arange(0,60,10).reshape(-1,1)+np.arange(0,6)
a

array([[ 0,  1,  2,  3,  4,  5],
       [10, 11, 12, 13, 14, 15],
       [20, 21, 22, 23, 24, 25],
       [30, 31, 32, 33, 34, 35],
       [40, 41, 42, 43, 44, 45],
       [50, 51, 52, 53, 54, 55]])

a[0,3:5]

array([3, 4])

a[4:,4:] #0开始

array([[44, 45],
       [54, 55]])

a[2::2,::2] #行从2开始，步长为2取，列从0开始，步长为2取

array([[20, 22, 24],
       [40, 42, 44]])

a[(0,1,2,3,4,5),(0,1,2,3,4,5)]

array([ 0, 11, 22, 33, 44, 55])

mask=np.array([1,0,1,0,0,1],dtype=np.bool)
a[mask,2]
a[mask, 2] : 下标的第0轴是一个布尔数组，它选取第0，2，5行；第1轴是一个整数，选取第2列。

array([ 2, 22, 52])

结构数组

假设我们需要定义一个结构数组，它的每个元素都有name, age和weight字段。在NumPy中可以如下定义：

#定义一下数组的结构类型
persontype = np.dtype({
    'names':['name', 'age', 'weight'],
    'formats':['S32','i', 'f']})

a = np.array([("Zhang",32,75.5),("Wang",24,65.2)],
    dtype=persontype)

我们先创建一个dtype对象persontype，通过其字典参数描述结构类型的各个字段。字典有两个关键字：names，formats。每个关键字对应的值都是一个列表。names定义结构中的每个字段名，而formats则定义每个字段的类型：

• S32 : 32个字节的字符串类型，由于结构中的每个元素的大小必须固定，因此需要指定字符串的长度
• i : 32bit的整数类型，相当于np.int32
• f : 32bit的单精度浮点数类型，相当于np.float32

然后我们调用array函数创建数组，通过关键字参数 dtype=persontype， 指定所创建的数组的元素类型为结构persontype。运行上面程序之后，我们可以在IPython中执行如下的语句查看数组a的元素类型

a.dtype

dtype([('name', 'S32'), ('age', '<i4'), ('weight', '<f4')])

这里我们看到了另外一种描述结构类型的方法： 一个包含多个组元的列表，其中形如 (字段名, 类型描述) 的组元描述了结构中的每个字段。类型描述前面为我们添加了 ‘|’, ‘<’ 等字符，这些字符用来描述字段值的字节顺序：

• | : 忽视字节顺序
• < : 低位字节在前
• > : 高位字节在前

结构数组的存取方式和一般数组相同，通过下标能够取得其中的元素，注意元素的值看上去像是组元，实际上它是一个结构：

a[0]

(b'Zhang', 32, 75.5)

a[0]['name']='li'
a[0]

(b'li', 32, 75.5)

我们不但可以获得结构元素的某个字段，还可以直接获得结构数组的字段，它返回的是原始数组的视图，因此可以通过修改b[0]改变a[0][“age”]：

b=a[:]['age']
b

array([32, 24], dtype=int32)

b[0]=40
a[0]['age']

40

内存对齐

C语言的结构体为了内存寻址方便，会自动的添加一些填充用的字节，这叫做内存对齐。例如如果把下面的name[32]改为name[30]的话，由于内存对齐问题，在name和age中间会填补两个字节，最终的结构体大小不会改变。因此如果numpy中的所配置的内存大小不符合C语言的对齐规范的话，将会出现数据错位。为了解决这个问题，在创建dtype对象时，可以传递参数align=True，这样numpy的结构数组的内存对齐和C语言的结构体就一致了。

#内存结构不是看很懂！！