Python Pandas CategoricalIndex – 新しいカテゴリを追加する方法

PandasのCategoricalIndexに新しいカテゴリを追加するには、add_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の表示

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

print("CategoricalIndex...\n",catIndex)

新しいカテゴリの追加

add_categories()メソッドを使用して新しいカテゴリを追加します。引数として新しいカテゴリのリストを渡します。追加されたカテゴリは、カテゴリ一覧の最後（最上位）に配置されます。

print("\nCategoricalIndex after adding new categories...\n",catIndex.add_categories(["a", "b", "c", "d"]))

完全なコード例

以下に完全なコードを示します。

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("CategoricalIndex...\n",catIndex)

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

# add_categories()を使用して新しいカテゴリを追加
# 新しいカテゴリはカテゴリ一覧の最後に追加される
print("\nCategoricalIndex after adding new categories...\n",catIndex.add_categories(["a", "b", "c", "d"]))

実行結果

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

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

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

CategoricalIndex after adding new categories...
CategoricalIndex(['p', 'q', 'r', 's', 'p', 'q', 'r', 's'], categories=['p', 'q', 'r', 's', 'a', 'b', 'c', 'd'], ordered=True, dtype='category')

まとめ

PandasのCategoricalIndexに新しいカテゴリを追加するには、add_categories()メソッドが便利です。出力結果からわかるように、元のデータ値はそのまま維持され、カテゴリ一覧に「a」「b」「c」「d」が末尾に追加されていることが確認できます。なお、既存のカテゴリと重複する値を追加しようとした場合はエラーになるため注意が必要です。