python学习记录

Posted 2023-02-21 Maggie张张

itertools.groupy(iterable, [key])

函数的作用：按照key对迭代器iterable进行分组
函数的返回：tuple，其中第一个元素是groupby后的键值，第二个元素是一个迭代器，包含那些被groupby的对象。
第二个参数key是可选的，如果缺省，那就按照iterable的每个元素分组。如果不缺省，那应该是一个函数，将iterable中的每个元素作用于这个函数。

#举个例子1
from itertools import groupby
L = ['aa','bb','ccc','ddd','csadfa']
for i,k in groupby(L,len):
    print i, list(k)

解读：
L是一个list迭代器，其中每个元素都是str；
groupby的第一个参数就是L, 第二个参数是个函数len，将len函数作用于L的每个元素，得到每个字符串的长度为2，2，3，3，6；
然后对这5个数字groupby，得到2，3，6这三个分组；其中，2的组中有’aa’, ‘bb’, 3的组中有’ccc’, ‘ddd’，6的组中有’csadfa’。
groupby的返回是tuple，for循环中i, k分别对应tuple的第0个和第1个元素，i为2时，k为2组中元素组成的迭代器，list(k)就把其中元素作为一个list输出，后两个一样。
输出：

2 ['aa', 'bb']
3 ['ccc', 'ddd']
6 ['csadfa']

#举个例子2
L = ['date' : 1,'date' : 2]
out = [(name, list(group)) for name, group in groupby(L, lambda p:p['date'])]
print out

解读：
L是一个list，其中的每个元素是dictionary
out是一个list，其中的每个元素是一个tuple
groupby的第一个元素是迭代器L，第二个元素是lambda表达式构建的匿名函数，将L的每个元素作为输入。因为L的每个元素p是字典，所以函数的返回值是以date为键值的value。所以groupby的分组是1，2，其中1的分组有元素‘date’:1, 2的分组中有元素‘date’:2
for循环中name为1时，group是‘date’:1，list(group)就是[‘date’:1]
输出：

[(1, ['date': 1]), (2, ['date': 2])]

map函数

map(func, iter1, iter2)，iter2可选, 也可以有更多的迭代器
通常接收一个函数和一个list迭代器，把函数作用于每一个迭代器中元素，然后返回一个新的list

#举个例子
L = [1, 2, 3, 4, 9]
print map(lambda x:x**2, L)
#返回 [1, 4, 9, 16, 81]

如果迭代器不只一个，那么前面的函数应该接收相应个数的参数，依次从这几个迭代器中取元素。

#举个例子2
L1 = [1, 2, 3, 4, 9]
L2 = [3, 10, 20, 30, 40]
L3 = [10, 20, 30, 40, 50]
print map(lambda a,b,c:a+b+c, L1, L2, L3)

#返回：[14, 32, 53, 74, 99]

enumerate函数

函数接受一个迭代器，然后返回若干元组，元组的第一个值是索引，第二个值是迭代器的元素。

L = [1, 2, 3, 4, 9]
for i, item in enumerate(L):
    print i, item
#返回：
0 1
1 2
2 3
3 4
4 9