python之array赋值技巧

首先上一段程序：

import numpy as np

list_a = list(range(10))
print("list_a: {}".format(list_a))

array_a = np.array(list_a)
print("array_a: {}".format(array_a))

list_b = [True,False,True,True,False,False,True,False,True,False]
print("list_b: {}".format(list_b))

array_b = np.array(list_b)
print("array_b: {}".format(array_b))

mask = np.zeros(10,dtype=np.bool)
print("mask: {}".format(mask))

下面是这段程序的输出：

list_a: [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
array_a: [0 1 2 3 4 5 6 7 8 9]
list_b: [True, False, True, True, False, False, True, False, True, False]
array_b: [ True False  True  True False False  True False  True False]
mask: [False False False False False False False False False False]

从输出可以看出list和array的不同之处是list是以逗号作为分隔符，而array是以空格作为分隔符。还有numpy中的zeros生成的是array。其实，list和array还有很多的不同之处，例如：

list是Python中的普通列表对象，支持append和attend操作，没有shape属性；array是numpy数据库中的对象，不支持append和attend操作，具有shape属性。

一个list中可以存放不同类型的数据，如int、float、str或者布尔型；而array中只能存放相同类型的数据。

list不支持乘法操作；array支持乘法操作。

而对于赋值操作，操作对象只能是array,而索引可以是array也可以是list：

d = array_a[list_b]
print("array_a[list_b]: {}".format(d))

e = array_a[array_b]
print("array_a[array_b]: {}".format(e))

mask[list_b] = 1
print("mask: {}".format(mask))

输出如下：

array_a[list_b]: [0 2 3 6 8]
array_a[array_b]: [0 2 3 6 8]
mask: [ True False  True  True False False  True False  True False]

此番操作可以将array_a中对应b数组或列表True位上的元素组成一个新的数组。

如果对列表进行如此赋值操作，即把程序中的array_a换成list_a，则会报错：

d = list_a[list_b]
TypeError: list indices must be integers or slices, not list

e = list_a[array_b]
TypeError: only integer scalar arrays can be converted to a scalar index