python - 基于python dict中相似键的Concat python值

Question

我有以下字典:

dct = { (1/1 , 2/1, 'AAPL') : pd.DataFrame({'aapl' : [1,2,3]}),
    (1/1, 2/1 , 'XOM') : pd.DataFrame({'xom' : [4,5,6]}) }

我想做的是根据键的前 2 个值连接数据帧(例如，如果 1/1 和 2/1 在键之间匹配，则连接这些帧)。

结果应该是这样的:

dct = { (1/1 , 2/1) : pd.DataFrame({'aapl' : [1,2,3], 'xom' : [4,5,6] })

Answer 1

defaultdict 是一种方法:

from collections import defaultdict
import pandas as pd

# initialize defaultdict with an empty data frame as default value
result = defaultdict(pd.DataFrame)
​
for k, v in dct.items():
    result[k[:2]] = pd.concat([result[k[:2]], v], axis=1)

result
#defaultdict(pandas.core.frame.DataFrame, {('1/1', '2/1'):    aapl  xom
#             0     1    4
#             1     2    5
#             2     3    6})