DataFrame к списку списков - PullRequest
       5

DataFrame к списку списков

0 голосов
/ 08 января 2020

У меня есть DataFrame df с несколькими columns, включая срок действия и iv. Я хотел бы создать lol = list-of-list [[]], чтобы каждая строка в lol соответствовала заданному сроку действия. Например, скажем df выглядит следующим образом:

df =


expiry             iv
===========================
2019-01-18,        .18
2019-01-18,        .17
2019-01-18,        .16
2019-02-18,        .14
2019-02-18,        .13
2019-02-18,        .12

lol будет выглядеть следующим образом :

lol = [
[0.18, 0.17, 0.16],
[0.14, 0.13, 0.12]]

Таким образом, каждая строка lol будет соответствовать соответствующей дате в df.

1 Ответ

2 голосов
/ 08 января 2020

IIU C

LoL=df.groupby('expiry').iv.agg(list).tolist()
LoL
[[0.18, 0.17, 0.16], [0.14, 0.13, 0.12]]
...