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