groupby.transform('count')の内容について

Question

DataFrameの重複行のサイズを調べる際にgroupby.transform('count')を用いて
サイズを求めることができたのですが、コードの意味が分からなかったため質問させていただきます。

使用したコードの例として

n=10
df = pd.DataFrame({
    'Rank':np.random.choice(['A','B','C'],n),
    'Score':np.random.randint(0,100,n)})

# Rankのサイズを'count' Columnに追加する
df['count'] = df.groupby('Rank')['Score'].transform('count')
print(df)

この際の['Score']の意味が分かりませんでした。
この部分を['Rank']に変えても同じ値を出力するため余計に分からなくなってしまいました。
良ければこの部分を意味を教えていただけると幸いです。
よろしくお願いします。

Answer 1

この際の['Score']の意味が分かりませんでした。
この部分を['Rank']に変えても同じ値を出力するため余計に分からなくなってしまいました。

このケースではとくに意味はありません。指定しなくてもよいです。
'count'はそのグループのサイズ（行数）です。よって列に関係なく同じ値になります。

いっぽうたとえば'sum'はそのグループ内の値の合計なので、列によって異なる値になります。

import pandas as pd

df = pd.DataFrame({'Rank':list('AAABBC'), 'Score':range(6)})
grp = df.groupby('Rank')['Score']
df['count'] = grp.transform('count')
df['sum'] = grp.transform('sum')
print(df)
"""
  Rank  Score  count  sum
0    A      0      3    3
1    A      1      3    3
2    A      2      3    3
3    B      3      2    7
4    B      4      2    7
5    C      5      1    5
"""

Comments

[alfald]

回答ありがとうございます。
そういう意味だったのですね。
とても参考になりました。