データフレーム2列をarray形式にした

Question

以下のデータフレームをarrayの形にしたいです。
ソートでID→rankの順で行い以下のarrayの要素でまとめたいです。
IDは同じ値ごと、rankは1～N、1～Nという区切りです。
どうすればよいかわからず止まってしまいました。
よろしくお願いします。

DFは以下

	id	rank    todo
0	15	1       10
1	15	2       16
2	15	3       10
3	15	1       11
4	15	2       14
5       16      1       11
6       16      2       20
7       16      3       13
7       16      1       16

→

 np.array([
        [10,16,10],
        [11,14],
        [11,20,13]
        [16]
        ])

Comments

[WathMorks]

質問の意味がわかりません。

np.arrayで４番目の[13,16]がペアになるのはなぜですか？

[退会済みユーザー]

こちら失礼しました。ご指摘通り、修正しました。有難うございます。

[meg_]

DataFrameは既に並び替えられているのですね？

[退会済みユーザー]

はい

Answer 1

申し訳ないです。勘違いしておりました。
通常通りのやり方で0をいれることできました。
失礼しました。

Answer 2

def groupSequence(lst): 
    lst1=lst[0]
    lst2=lst[1]
    res = [[lst2[0]]] 
    
    for i in range(1, len(lst1)): 
        if lst1[i-1]+1 == lst1[i]: 
            res[-1].append(lst2[i]) 
  
        else: 
            res.append([lst2[i]]) 
    return res

ID=[15,15,15,15,15,16,16,16,16]
rank=[1,2,3,1,2,1,2,3,1]
todo=[10,16,10,11,14,11,20,13,16]
df=pd.DataFrame({'ID':ID,'rank':rank,  'todo':todo})

df.groupby('ID').agg(lambda x: list(x)).apply(groupSequence,axis=1).agg(sum)

Comments

[退会済みユーザー]

こちら大変助かりました。
丁寧に有難うございます。

[退会済みユーザー]

ちなみに、このarrayを最長の列の長さで統一するために、15列に足りない行には0を代入したいのですが可能でしょうか。
どうしても、データ型のせいか、うまく0もしくはnanにすることができません。よろしくお願いします。