python 中字典使用的一些小技巧

本文簡單的記錄了 python中 字典使用的一些技巧

一、從字典中提取子集

prices = {
'ACME': 45.23,
'AAPL': 612.78,
'IBM': 205.55,
'HPQ': 37.20,
'FB': 10.75
}

##下面使用字典推導式來取 值大於200或者鍵在一個列表中的子集

## Make a dictionary of all prices over 200
p1 = {key: value for key, value in prices.items() if value > 200}

tech_names = {'AAPL', 'IBM', 'HPQ', 'MSFT'}
p2 = {key: value for key, value in prices.items() if key in tech_names}

二、當數據框的一列爲字典時，可以嘗試用apply函數將其拆分爲多個列

##數據框的第二列 是一個字典

df = pd.DataFrame({'a':[1,2,3], 'b':[{'c':1}, {'d':3}, {'c':5, 'd':6}]})

df
Out[72]: 
   a                 b
0  1          {'c': 1}
1  2          {'d': 3}
2  3  {'c': 5, 'd': 6}

## 通過apply函數將第二列拆分爲 c、d列。

df['b'].apply(pd.Series)
Out[73]: 
     c    d
0  1.0  NaN
1  NaN  3.0
2  5.0  6.0