データの列名の要素ごとのカウント方法を教えていただきたいです。

Question

pythonでの質問です！
下の「自分で作成したコード」で一応目的の結果が得られたのですが、ゴリ押しのような気がしてしまいます。「おしいコード」でもできそうですがなかなか難しい、、
「自分で作成したコード」の結果を変えずに、コンパクトに表現する方法はありますでしょうか。ご回答よろしくお願いします！

import pandas as pd
import numpy as np

df = pd.DataFrame({
    'name': ['Alice', 'Bob', 'Charlie', 'Dave', 'Ellen', 'Frank'],
    'age': [24, 42, 18, 68, 24, 30],
    'state': ['NY', 'CA', 'CA', 'TX', 'CA', 'NY'],
    'point': [64, 92, 70, 70, 88, 57]
})
df

＃自分で作成したコード
df_count = df.groupby('state').count()
df_count = df_count.iloc[:,range(1)]
df_count = df_count.set_axis(['count'], axis=1)
df_count

#おしいコード
df_count = df.groupby('state').size()
df_count

Answer 1

Named aggregationを使うと下記のように書けます。
countという名前の列を、state列をcountで集計して作る、という意味です。

df_count = df.groupby('state').agg(count=('state', 'count'))

別のやり方として、要素数を数えたいなら、そのための関数(value_counts)を使うほうが分かりやすいと思います。
groupbyではキー名、value_countsでは出現数でソートされるという違いがあります。

追記
DataFrameでvalue_countsが使えるのは 1.1.0以降の比較的新しめのpandasのようです。
列を抜き出して、そのSeriesにvalue_counts()すればいいです。

df_count = df['state'].value_counts().to_frame('count')
# df_count = df['state'].value_counts().sort_index().to_frame('count')

# pandas 1.1以降のみ
df_count = df.value_counts('state').to_frame('count')

Comments

[退会済みユーザー]

回答ありがとうございます！
別のやり方で、'DataFrame' object has no attribute 'value_counts'というエラーがでてきてしまうのはどうしたらよいでしょうか。

[bsdfan]

pandasのバージョンが1.1.0より前だと、DataFrameでvalue_countsが使えないようです。
回答を修正しました。

[退会済みユーザー]

バージョンの問題だったのですね。修正コードもありがとうございました！

Answer 2

df_count = df.groupby('state').apply(len).to_frame('count')
#df_count = df.groupby('state').count()['name'].to_frame('count')

df_count
#
       count
state       
CA         3
NY         2
TX         1

Comments

[退会済みユーザー]

回答ありがとうございます！参考にさせていただきます！

Answer 3

一行で書くなら以下のようになります。

>>> df_count = df.groupby('state')[['state']].count().rename(columns={'state':'count'})
>>> df_count
       count
state
CA         3
NY         2
TX         1

Comments

[退会済みユーザー]

回答ありがとうございます！参考にさせていただきます！