Python中Array的常用操作数组高级操作

Posted 2021-06-03 多鱼的夏天

1. 抽取数组中不重复的数据

>>> a = [ x*2 for x in range(1,5) ]*2

>>> uniq = list(set(a))

[8, 2, 4, 6]

>>> uniq = sorted(set(a))

[2, 4, 6, 8]

>>> b={}

>>> for x in a:

… b[x]=1

>>> uniq = b.keys()

 

2. 去除在另一个数组中元素

>>> a = [ x*2 for x in range(1,5) ]

>>> b = [ x for x in a if x >3 ]

 

>>> aonly = [x for x in a if x not in b]

 

>>> a_set = set(a)

>>> b_set = set(b)

>>> aonly = list(a_set - b_set)

3. 数组排序

>>> def comp(x,y):

… if x>y:

… return -1

… elif x==y:

… return 0

… else :

… return 1

>>> unsorted_list = [82, 67, 10, 46, 81, 40, 71, 88, 55]

>>> unsorted_list.sort(comp)

[88, 82, 81, 71, 67, 55, 46, 40, 10]

 

4. 两个数组交、并操作

 

>>> a = [ x*2 for x in range(1,5) ]

[2, 4, 6, 8]

>>> b = [ x for x in range(3,7) ]

[3, 4, 5, 6]

>>> a_set = set(a)

>>> b_set = set(b)

# Union

>>> print list (a_set | b_set )

[2, 3, 4, 5, 6, 8]

#Intersection

>>> print list(a_set & b_set)

[4, 6]

#Difference

>>> print list(a_set ^ b_set)

[8, 2, 3, 5]

5. 数组函数map()、filter()、reduce()

a) map()

map(function,sequence)为每一个sequence的元素调用function(item) 并把返回值组成一个数组。

>>>

>>> def fun(x): return x*x

>>> print map(fun,range(0,5))

[0, 1, 4, 9, 16]

使用map(None,list1,list2)可以快速把两个数组变成元素对的一个数组

>>> print map(None,range(0,5),range(100,105))

[(0, 100), (1, 101), (2, 102), (3, 103), (4, 104)]

b) filter()

filter(function,sequence)返回一个序列，包含了所有调用function(item)后返回值为true的元素。

>>> unsorted_list = [82, 67, 10, 46, 81, 40, 71, 88, 55]

>>> def fun(x): return x%2==0

>>> print filter(fun,unsorted_list)

[88, 82, 46, 40, 10]

c) reduce()

reduce(function,sequence)返回一个单值，首先以数组的前两个元素调用函数function，在以返回值和第三个元素为参数调用，依次执行下去。

例如，以下程序计算1到10的整数的和

>>> def add(x,y): return x+y

>>> print reduce(add,range(0,11))

55