玩轉python字典與列表（下）

本文首發於知乎

本文分爲以下部分

• 列表轉置
• 字典轉置
• 字典列表之間的轉換
• 總結

列表轉置

已有

l = [['a', 'b', 'c'],
     [1, 2, 3]]
複製代碼

想獲得

[('a', 1), ('b', 2), ('c', 3)]
複製代碼

代碼以下

[z for z in zip(*l)]
複製代碼

zip函數在list的轉置中充當重要做用，它的基礎用法是這樣的

for z in zip(['a', 'b', 'c'], [1, 2, 3]):
    print(z)

# 輸出
('a', 1)
('b', 2)
('c', 3)
複製代碼

字典的轉置

已有

m = {'Bob': {'age': 30, 'country': 'America'},
     'Mary': {'age': 20, 'country': 'China'},
     'Frank': {'age': 25, 'country': 'America'}}
複製代碼

想獲得

{'age': {'Bob': 30, 'Frank': 25, 'Mary': 20},
 'country': {'Bob': 'America', 'Frank': 'America', 'Mary': 'China'}}
複製代碼

若是借用pandas直接轉置便可

>>> m = {'Bob': {'age': 30, 'country': 'America'},
...      'Mary': {'age': 20, 'country': 'China'},
...      'Frank': {'age': 25, 'country': 'America'}}
>>>
>>> import pandas as pd
>>> pd.DataFrame(m)
             Bob    Frank   Mary
age           30       25     20
country  America  America  China
>>> pd.DataFrame(m).transpose()
      age  country
Bob    30  America
Frank  25  America
Mary   20    China
>>> pd.DataFrame(m).transpose().to_dict()
{'age': {'Mary': 20, 'Bob': 30, 'Frank': 25}, 'country': {'Mary': 'China', 'Bob': 'America', 'Frank': 'America'}}
複製代碼

若是咱們單獨使用字典的知識也是能夠辦到的

from collections import defaultdict
result = defaultdict(dict)
for names, infos in m.items():
    for k, v in infos.items():
        result[k].update({names: v})
        
dict(result)
複製代碼

字典列表之間的轉換

已有

d = {'name': ['a', 'b', 'c'],
    'num': [1, 2, 3]}
複製代碼

想要獲得

[('a', 1), ('b', 2), ('c', 3)]

[{'name': 'a', 'num': 1}, 
 {'name': 'b', 'num': 2},
 {'name': 'c', 'num': 3}]
複製代碼

兩種形式

代碼以下

# 第一個
[z for z in zip(*d.values())]

# 第二個
[dict(zip(d.keys(), z)) for z in zip(*d.values())]
# 或
[dict(zip(d, z)) for z in zip(*d.values())]
複製代碼

總結

首先來看一下咱們遇到了哪些形式的數據

{'name' : ['a', 'b', 'c'],
 'num' : [1, 2, 3]}
 
{'Bob': {'age': 30, 'country': 'America'},
 'Mary': {'age': 20, 'country': 'China'},
 'Frank': {'age': 25, 'country': 'America'}}
 
{'age': {'Bob': 30, 'Frank': 25, 'Mary': 20},
 'country': {'Bob': 'America', 'Frank': 'America', 'Mary': 'China'}}
 
[('a', 1), ('b', 2), ('c', 3)]
[{'a', 1}, {'b', 2}, {'c', 3}]
[{'a': 1}, {'b': 2}, {'c': 3}]
[{'name': 'Bob', 'age': 30, 'country': 'America'},
 {'name': 'Mary', 'age': 20, 'country': 'China'},
 {'name': 'Frank', 'age': 25, 'country': 'America'}]
複製代碼

下面總結一下看待字典列表的思想

• 首先，最重要的是列表這種結構，將集合、元組都看作列表（無序等都是細枝末節）
• 而字典能夠看作特殊的列表，即對每一個元素都命名了的列表，鍵是名，值至關於列表的元素

讀者能夠試着使用這種思想去看上面這些數據，就會明白，爲何某些數據直接能夠用pd.DataFrame轉化爲數據框，他們都至關於二維列表造成的矩陣。一個維度，若是以列表形式呈現則視爲行，若是以字典形式呈現則視爲列。

這些數據的相互轉化中，只要根據要生成結果的形式，記住幾個套路便可：

1.若是結果是一個字典，簡單的用字典生成式等方式不談。當值是動態更新的時候，則使用collections模塊中的defaultdict設置值的類型，使用循環填入數據，若是值類型是list則用append方法填入，若是是dict則用update方法填入。

注：有時結果是字典會直接人爲指定鍵，值用列表生成式等直接得到，這是一種比較不通用的作法，由於須要人爲指定鍵的話，就沒法將其改裝爲一個適合各類狀況的函數了。

2.若是結果是一個列表，則用列表生成式，構造每個元素的形式便可

3.若是涉及到了轉置問題，其實本質上都是列表的轉置，往zip函數那裏想，有時也能夠用defaultdict

歡迎關注個人知乎專欄

專欄主頁：python編程

專欄目錄：目錄

版本說明：軟件及包版本說明