Python字典:collections 中 defaultdict 的使用

水深无声 2024-03-30 10:48 61阅读 0赞

#### 1.字典的键映射多个值 ####

将下面的列表转成字典

> l = \[('a',2),('b',3),('a',1),('b',4),('a',3),('a',1),('b',3)\]

一个字典就是一个键对应一个单值的映射，而上面的列表中有相同键。如果你想要一个键映射多个值，那么就需要将这多个值放到另外的序列中，比如 list 或者 set 里面，像下面这样:

你可以很方便的使用 collections 模块中的 defaultdict 来构造这样的字典。 defaultdict 的一个特征是它会自动初始化每个 key 刚开始对应的值。

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 4 5 6 7 8 </td> 
 <td> <code>In [</code><code>1</code><code>]: l </code><code>=</code> <code>[(</code><code>'a'</code><code>,</code><code>2</code><code>),(</code><code>'b'</code><code>,</code><code>3</code><code>),(</code><code>'a'</code><code>,</code><code>1</code><code>),(</code><code>'b'</code><code>,</code><code>4</code><code>),(</code><code>'a'</code><code>,</code><code>3</code><code>),(</code><code>'a'</code><code>,</code><code>1</code><code>),(</code><code>'b'</code><code>,</code><code>3</code><code>)]</code> <code>In [</code><code>2</code><code>]: </code><code>from</code> <code>collections </code><code>import</code> <code>defaultdict</code> <code>In [</code><code>3</code><code>]: d </code><code>=</code> <code>defaultdict(</code><code>list</code><code>)</code> <code>In [</code><code>4</code><code>]: </code><code>for</code> <code>key, value </code><code>in</code> <code>l:</code> <code>&nbsp;&nbsp;&nbsp;</code><code>...:&nbsp;&nbsp;&nbsp;&nbsp; d[key].append(value)</code> <code>&nbsp;&nbsp;&nbsp;</code><code>...: </code> <code>In [</code><code>5</code><code>]: d</code> <code>Out[</code><code>5</code><code>]: defaultdict(</code><code>list</code><code>, { </code><code>'a'</code><code>: [</code><code>2</code><code>, </code><code>1</code><code>, </code><code>3</code><code>, </code><code>1</code><code>], </code><code>'b'</code><code>: [</code><code>3</code><code>, </code><code>4</code><code>, </code><code>3</code><code>]})</code> </td> 
 </tr> 
 </tbody> 
</table>

当然这个默认的容器不一定是 list, 也可以是集合 set。根据自己的需求选择用 list 还是 set 。如果你想保持元素的插入顺序就应该使用列表，如果想去掉重复元素就使用集合！

#### 2.统计字典中某个值出现的次数 ####

来源于微信交流群里一个朋友工作中的问题，列表中有很多字典，需要统计字典中相同的键对应的值的和

利用 defaultdict 设置默认值的方法 defaultdict(int)，代码如下:

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 4 5 6 7 8 </td> 
 <td> <code>In [</code><code>6</code><code>]: d </code><code>=</code> <code>defaultdict(</code><code>int</code><code>)</code> <code>In [</code><code>7</code><code>]: objs </code><code>=</code> <code>[{ </code><code>'F29958SVDK6'</code><code>: </code><code>12</code><code>}, { </code><code>'F29958SVDK6'</code><code>: </code><code>12</code><code>}, { </code><code>'F29958SVDK6'</code><code>: </code><code>12</code><code>}, { </code><code>'F29958SVDK6'</code><code>: </code><code>12</code><code>}, { </code><code>'F29958SVDK6'</code><code>: </code><code>12</code><code>}]</code> <code>In [</code><code>8</code><code>]: </code><code>for</code> <code>obj </code><code>in</code> <code>objs:</code> <code>&nbsp;&nbsp;&nbsp;</code><code>...:&nbsp;&nbsp;&nbsp;&nbsp; </code><code>for</code> <code>key,value </code><code>in</code> <code>obj.items():</code> <code>&nbsp;&nbsp;&nbsp;</code><code>...:&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp; d[key] </code><code>+</code><code>=</code> <code>value</code> <code>&nbsp;&nbsp;&nbsp;</code><code>...: </code> <code>In [</code><code>9</code><code>]: d</code> <code>Out[</code><code>9</code><code>]: defaultdict(</code><code>int</code><code>, { </code><code>'F29958SVDK6'</code><code>: </code><code>60</code><code>})</code> </td> 
 </tr> 
 </tbody> 
</table>

### 二、collections 创建有序字典 ###

字典dict是无序的，如果我们想要有序的dict，可以使用OrdereDict 。示例如下：

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 4 5 6 7 </td> 
 <td> <code>In [</code><code>11</code><code>]: </code><code>from</code> <code>collections </code><code>import</code> <code>OrderedDict</code> <code>In [</code><code>12</code><code>]: d </code><code>=</code> <code>OrderedDict()</code> <code>In [</code><code>13</code><code>]: d[</code><code>'bar'</code><code>] </code><code>=</code> <code>2</code> <code>In [</code><code>14</code><code>]: d[</code><code>'non'</code><code>] </code><code>=</code> <code>8</code> <code>In [</code><code>15</code><code>]: d[</code><code>'sek'</code><code>] </code><code>=</code> <code>5</code> <code>In [</code><code>16</code><code>]: d</code> <code>Out[</code><code>17</code><code>]: OrderedDict([(</code><code>'bar'</code><code>, </code><code>2</code><code>), (</code><code>'non'</code><code>, </code><code>8</code><code>), (</code><code>'sek'</code><code>, </code><code>5</code><code>)])</code> </td> 
 </tr> 
 </tbody> 
</table>

OrderedDict 内部维护着一个根据键插入顺序排序的双向链表。每次当一个新的元 素插入进来的时候，它会被放到链表的尾部。对于一个已经存在的键的重复赋值不会 改变键的顺序。

需要注意的是，一个 OrderedDict 的大小是一个普通字典的两倍，因为它内部维 护着另外一个链表。所以如果你要构建一个需要大量 OrderedDict 实例的数据结构的 时候 (比如读取 100,000 行 CSV 数据到一个 OrderedDict 列表中去)，那么你就得仔细权衡一下是否使用 OrderedDict 带来的好处要大过额外内存消耗的影响。

#### 1.改变 key-value 的顺序 ####

OrderedDict 是有序的字典，同时也能改变其顺序。比如我们想要改变有序的 OrderedDict 对象的 key-value 顺序，可以使用 `move_to_end(key)`。还是以上面创建的有序字典为例子

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 </td> 
 <td> <code>In [</code><code>18</code><code>]: d.move_to_end(</code><code>"bar"</code><code>)</code> <code>In [</code><code>20</code><code>]: d</code> <code>Out[</code><code>20</code><code>]: OrderedDict([(</code><code>'non'</code><code>, </code><code>8</code><code>), (</code><code>'sek'</code><code>, </code><code>5</code><code>), (</code><code>'bar'</code><code>, </code><code>2</code><code>)])</code> </td> 
 </tr> 
 </tbody> 
</table>

可以看到之前排在第一位的 bar被移到最后一位了。`move_to_end`还接收一个关键字参数 last。last 默认为 True，当 last = False 的时候，表示将该键移动到最前面！

#### 2.删除 key\_value ####

如果我们要删除有序字典中的 key-value, 可以使用 popitem 方法， popitem(last=True) 按照先进后出的顺序删除 dict中 的 key-value，popitem(last=False) 按照先进先出的规则删除 dict 中的 key-value。

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 4 5 6 </td> 
 <td> <code>In [</code><code>42</code><code>]: d</code> <code>Out[</code><code>42</code><code>]: OrderedDict([(</code><code>'bar'</code><code>, </code><code>2</code><code>), (</code><code>'non'</code><code>, </code><code>8</code><code>), (</code><code>'sek'</code><code>, </code><code>5</code><code>)])</code> <code>In [</code><code>43</code><code>]: d.popitem(last</code><code>=</code><code>False</code><code>)</code> <code>Out[</code><code>43</code><code>]: (</code><code>'bar'</code><code>, </code><code>2</code><code>)</code> <code>In [</code><code>44</code><code>]: d</code> <code>Out[</code><code>44</code><code>]: OrderedDict([(</code><code>'non'</code><code>, </code><code>8</code><code>), (</code><code>'sek'</code><code>, </code><code>5</code><code>)])</code> </td> 
 </tr> 
 </tbody> 
</table>

来源：[https://www.weidianyuedu.com][https_www.weidianyuedu.com]

[https_www.weidianyuedu.com]: https://www.weidianyuedu.com