Python Pandas – 2つのCategoricalIndexオブジェクトに同じ要素が含まれるかどうかを判定する方法

Pandasで2つのCategoricalIndexオブジェクトに同じ要素が含まれているかどうかを確認するには、equals()メソッドを使用します。このメソッドは、2つのインデックスの要素・並び順・カテゴリの定義・順序情報(ordered)がすべて一致する場合にTrueを返し、そうでなければFalseを返します。

CategoricalIndexの作成と比較の手順

まず、必要なライブラリをインポートします。

import pandas as pd

「categories」パラメータを使ってカテゴリカルデータのカテゴリを設定し、「ordered」パラメータを使って順序付きカテゴリとして扱います。その上で、2つのCategoricalIndexオブジェクトを作成します。

catIndex1 = pd.CategoricalIndex(['p', 'q', 'r', 's', 'p', 'q', 'r', 's'], ordered=True, categories=['p', 'q', 'r', 's'])
catIndex2 = pd.CategoricalIndex(['p', 'q', 'r', 's', 'p', 'q', 'r', 's'], ordered=True, categories=['p', 'q', 'r', 's'])

equals()メソッドを呼び出して、両方のCategoricalIndexオブジェクトが等しいかどうかを確認します。

print('\n2つのCategoricalIndexオブジェクトが等しいかどうかを確認...\n', catIndex1.equals(catIndex2))

equals()メソッドのポイント

• 要素の値だけでなく、並び順も一致している必要があります。
• categories(カテゴリの定義)が同一であることも比較対象になります。
• ordered(順序付きかどうか)の設定が異なる場合も、等しいとは判定されません。

サンプルコード(全体)

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

import pandas as pd

# 「categories」パラメータでカテゴリを設定し、「ordered」パラメータで順序付きとして扱う
# 2つのCategoricalIndexオブジェクトを作成する
catIndex1 = pd.CategoricalIndex(['p', 'q', 'r', 's', 'p', 'q', 'r', 's'], ordered=True, categories=['p', 'q', 'r', 's'])
catIndex2 = pd.CategoricalIndex(['p', 'q', 'r', 's', 'p', 'q', 'r', 's'], ordered=True, categories=['p', 'q', 'r', 's'])

# CategoricalIndexオブジェクトを表示
print('CategoricalIndex1...\n', catIndex1)
print('\nCategoricalIndex2...\n', catIndex2)

# 2つのCategoricalIndexオブジェクトが等しいかどうかを確認
print('\n2つのCategoricalIndexオブジェクトが等しいかどうかを確認...\n', catIndex1.equals(catIndex2))

実行結果

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

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

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

2つのCategoricalIndexオブジェクトが等しいかどうかを確認...
True

このように、2つのCategoricalIndexオブジェクトがまったく同じ要素・カテゴリ・順序設定を持っているため、equals()メソッドはTrueを返しました。片方でも要素やカテゴリの定義が異なる場合はFalseとなるため、データの一致検証に便利なメソッドです。