データフレーム内の要素でソート

Question

入力を例えば以下のようにしたとき，

list1=[[1,set('A')],[2,set(['B','C'])],[3,set(['A','B','C'])]]
columns1 =["number","set"]
pd.DataFrame(data=list1,columns=columns1)

このデータフレームをソートする場合，setの要素数の降順にソートするにはどのようにすればよいのでしょうか．

Answer 1

以下の手順でできます。

1. df["set"] で列 set を Series で取得する。
2. Series.map(len) で列 set の各要素数を表す Series を作成する。
3. Series.argsort() でソート後のインデックスを取得する。
4. argsort は昇順ソートなので、iloc[::-1] で降順ソートした結果にする。
5. df.iloc[sorted_idx] で元の DataFrame のソートした結果を取得する。

import pandas as pd

data = [[1, set("A")], [2, set(["B", "C"])], [3, set(["A", "B", "C"])]]
columns = ["number", "set"]
df = pd.DataFrame(data, columns=columns)
print(df)
#    number        set
# 0       1        {A}
# 1       2     {C, B}
# 2       3  {C, A, B}

sorted_idx = df["set"].map(len).argsort().iloc[::-1]
df = df.iloc[sorted_idx]
print(df)
#    number        set
# 2       3  {C, A, B}
# 1       2     {C, B}
# 0       1        {A}

Comments

[mermer]

ありがとうございます！