目录 | 上一节 (2.4 序列) | [下一节 (2.6 列表推导式)]()python
collections
模块为数据处理提供了许多有用的对象。本部分简要介绍其中的一些特性。git
假设要把每只股票的总份额表格化。github
portfolio = [ ('GOOG', 100, 490.1), ('IBM', 50, 91.1), ('CAT', 150, 83.44), ('IBM', 100, 45.23), ('GOOG', 75, 572.45), ('AA', 50, 23.15) ]
此表中有两个 IBM
条目,两个 GOOG
条目,它们应该以某种方式合并到一块儿。segmentfault
解决方案:使用 Counter
模块。bash
from collections import Counter total_shares = Counter() for name, shares, price in portfolio: total_shares[name] += shares total_shares['IBM'] # 150
问题:把一个键映射到多个值。app
portfolio = [ ('GOOG', 100, 490.1), ('IBM', 50, 91.1), ('CAT', 150, 83.44), ('IBM', 100, 45.23), ('GOOG', 75, 572.45), ('AA', 50, 23.15) ]
像以前的示例那样,键 IBM
应具备两个不一样的元组。翻译
解决方案:使用 defaultdict
模块。code
from collections import defaultdict holdings = defaultdict(list) for name, shares, price in portfolio: holdings[name].append((shares, price)) holdings['IBM'] # [ (50, 91.1), (100, 45.23) ]
defaultdict
模块确保每次访问键的时候获取到一个默认值。对象
问题:咱们须要最近 N 件事的历史。教程
解决方案:使用 deque
模块。
from collections import deque history = deque(maxlen=N) with open(filename) as f: for line in f: history.append(line) ...
collections
多是最有用的库模块之一,用于解决特殊用途的数据处理问题,例如表格化或者索引化。
在本练习中,咱们来看几个简单的例子。首先运行report.py
,以便在交互模式下可以加载股票投资组合。
bash % python3 -i report.py
假设须要将每支股票的份额总数表格化,那么使用 Counter
对象会很容易。试试看:
>>> portfolio = read_portfolio('Data/portfolio.csv') >>> from collections import Counter >>> holdings = Counter() >>> for s in portfolio: holdings[s['name']] += s['shares'] >>> holdings Counter({'MSFT': 250, 'IBM': 150, 'CAT': 150, 'AA': 100, 'GE': 95}) >>>
仔细观察portfolio
中的 MSFT
和 IBM
的多个条目是如何合并的。
能够像字典同样使用 Counter 模块检索单个值。
>>> holdings['IBM'] 150 >>> holdings['MSFT'] 250 >>>
若是想要对值排名,这样作:
>>> # Get three most held stocks >>> holdings.most_common(3) [('MSFT', 250), ('IBM', 150), ('CAT', 150)] >>>
让咱们获取另外一个股票投资组合并生成一个新的 Counter 对象:
>>> portfolio2 = read_portfolio('Data/portfolio2.csv') >>> holdings2 = Counter() >>> for s in portfolio2: holdings2[s['name']] += s['shares'] >>> holdings2 Counter({'HPQ': 250, 'GE': 125, 'AA': 50, 'MSFT': 25}) >>>
最后,经过一个简单的操做把全部的 holdings 变量合并。
>>> holdings Counter({'MSFT': 250, 'IBM': 150, 'CAT': 150, 'AA': 100, 'GE': 95}) >>> holdings2 Counter({'HPQ': 250, 'GE': 125, 'AA': 50, 'MSFT': 25}) >>> combined = holdings + holdings2 >>> combined Counter({'MSFT': 275, 'HPQ': 250, 'GE': 220, 'AA': 150, 'IBM': 150, 'CAT': 150}) >>>
这只是对 Counter 功能的一个小尝试,若是发现须要对值进行表格化,那么就应该考虑使用它。
collections
模块是 Python 全部库中最有用的库模块之一。实际上,咱们能够为此作一个拓展教程,可是,如今这样作会分散注意力。从如今开始,把collections
列为您的睡前读物,以备后用。
目录 | 上一节 (2.4 序列) | [下一节 (2.6 列表推导式)]()