collections扩展基本数据类型

collections 官方文档：
https://docs.python.org/zh-cn/3.7/library/collections.html

collections提供了加强版的数据类型，以下列举三个常用基本数据类型：

• namedtuple ——带命名的元组
• Counter ——计数器
• deque ——双向队列

 

namedtuple

使用namedtuple定义元组中存放的数据为Point点的x,y坐标。

支持通过位置顺序传递参数以及关键字传递参数。

#命名元组
import collections
#namedtuple带命名的元组，将x,y作为属性
Point = collections.namedtuple('Point', ['x', 'y'])
p = Point(11, y=22)
print(p[0] + p[1])
x, y =p
print(p.x, p.y)
print(p)

#33
#11 22
#Point(x=11, y=22)

扩展：将Point放到空间中，并计算两点间的距离（欧式距离）：

#计算欧式距离
from collections import namedtuple
from math import sqrt
Point = namedtuple('Point', ['x', 'y', 'z'])

class Vector(Point):
    def __init__(self, p1, p2, p3):
        super(Vector).__init__()
        self.p1 = p1
        self.p2 = p2
        self.p3 = p3

    def __sub__(self, other):
        tmp = (self.p1 - other.p1)**2+(self.p2 - other.p2)**2+(self.p3 - other.p3)**2
        return sqrt(tmp)
p1 = Vector(1,2,3)
p2 = Vector(4,5,6)
p1-p2
# 5.196152422706632

 

Counter

通过Counter将字符串传入，使用most_common(3)即可取出频率最高的三个值。

from collections import Counter
mystring = ['a', 'b', 'c', 'd', 'd', 'd', 'd', 'c', 'c', 'e']
#取频率最高的三个值
cnt = Counter(mystring)
cnt.most_common(3)
# [('d', 4), ('c', 3), ('a', 1)]
cnt['b']
# 1

deque

实现双向队列插入操作。

#双向队列
from collections import deque
d = deque('uvw')
d.append('xyz')
d.appendleft('abc')
# deque(['abc', 'u', 'v', 'w', 'xyz'])