Python Pandas CategoricalIndex – カテゴリの取得方法をわかりやすく解説

Pandas の CategoricalIndex に含まれるカテゴリを取得するには、categories プロパティを使用します。まずは、必要なライブラリをインポートしましょう。

import pandas as pd

CategoricalIndex は、限られた（通常は固定された）数の取り得る値、つまり「カテゴリ」のみを持つことができます。「categories」パラメータを使うことで、カテゴリカルデータに対してカテゴリを設定できます。また、「ordered」パラメータを指定すると、そのカテゴリカルデータを順序付きとして扱うことができます。

catIndex = pd.CategoricalIndex(["p", "q", "r", "s", "p", "q", "r", "s"], ordered=True, categories=["p", "q", "r", "s"])

作成した CategoricalIndex を表示してみましょう。

print("Categorical Index...\n", catIndex)

続いて、categories プロパティを使ってカテゴリを取得します。

print("\nDisplaying Categories from CategoricalIndex...\n", catIndex.categories)

コード例全体

以下に、ここまでの処理をまとめた完全なコードを示します。

import pandas as pd

# CategoricalIndex は限られた（通常は固定された）数の値（カテゴリ）のみを取ることができる
# 「categories」パラメータでカテゴリを設定
# 「ordered」パラメータで順序付きカテゴリとして扱う
catIndex = pd.CategoricalIndex(["p", "q", "r", "s", "p", "q", "r", "s"], ordered=True, categories=["p", "q", "r", "s"])

# CategoricalIndex を表示
print("Categorical Index...\n", catIndex)

# カテゴリを取得
print("\nDisplaying Categories from CategoricalIndex...\n", catIndex.categories)

実行結果

上記のコードを実行すると、次のような出力が得られます。

Categorical Index...
CategoricalIndex(['p', 'q', 'r', 's', 'p', 'q', 'r', 's'], categories=['p', 'q', 'r', 's'], ordered=True, dtype='category')

Displaying Categories from CategoricalIndex...
Index(['p', 'q', 'r', 's'], dtype='object')

出力を見ると、CategoricalIndex 自体には重複を含む値が格納されていますが、categories プロパティを参照することで、定義された一意なカテゴリの一覧（この場合は p、q、r、s）だけを抽出できることが確認できます。順序付き（ordered=True）として設定しているため、これらのカテゴリ間には大小関係が存在し、比較演算にも利用可能です。