Cách chuyển đổi khung dữ liệu gấu trúc sang từ điển phân cấp

16

Tôi có khung dữ liệu gấu trúc sau đây:

df1 = pd.DataFrame({'date': [200101,200101,200101,200101,200102,200102,200102,200102],'blockcount': [1,1,2,2,1,1,2,2],'reactiontime': [350,400,200,250,100,300,450,400]})

Tôi đang cố gắng tạo một từ điển phân cấp, với các giá trị của từ điển được nhúng dưới dạng danh sách, trông giống như sau:

{200101: {1:[350, 400], 2:[200, 250]}, 200102: {1:[100, 300], 2:[450, 400]}}

Làm thế nào tôi sẽ làm điều này? Gần nhất tôi nhận được là sử dụng mã này:

df1.set_index('date').groupby(level='date').apply(lambda x: x.set_index('blockcount').squeeze().to_dict()).to_dict()

Trả về:

{200101: {1: 400, 2: 250}, 200102: {1: 300, 2: 400}}

python pandas

— alechay
nguồn

20

Đây là một cách khác để sử dụng pivot_table:

d = df1.pivot_table(index='blockcount',columns='date',
     values='reactiontime',aggfunc=list).to_dict()

print(d)

{200101: {1: [350, 400], 2: [200, 250]},
 200102: {1: [100, 300], 2: [450, 400]}}

— anky
nguồn

7

IIUC

    df1.groupby(['date','blockcount']).reactiontime.agg(list).unstack(0).to_dict()
{200101: {1: [350, 400], 2: [200, 250]}, 200102: {1: [100, 300], 2: [450, 400]}}

— YOBEN_S
nguồn

5

Bạn có thể làm như sau

df2 = df1.groupby(['date', 'blockcount']).agg(lambda x: pd.Series(x).tolist())

# Formatting the result to the correct format
dct = {}
for k, v in df2["reactiontime"].items():
  if k[0] not in dct: 
    dct[k[0]] = {}
  dct[k[0]].update({k[1]: v})

Sản xuất,

>>> {200101: {1: [350, 400], 2: [200, 250]}, 200102: {1: [100, 300], 2: [450, 400]}}

dct giữ kết quả bạn cần

— thushv89
nguồn