Pandas DataFrameの比較(df.equals)が上手くいかない。csvファイルから読んだ個別要素がnumpy.int32と型が違う為か

Question

Pandas DataFrameで作ったデータをファイルに保存して、それを読み出してプログラム内のDataFrameと比較し、一致するかを調べようとしています。

下記コードでセーブした同じデータを比較する。
printされた限り同じデータに見えるのですが、
df.equals(d2)として比較するとFalseとなります。
中の要素を見てみるとプログラム中のDataFrameの中の個別要素はintで、csvファイルから読んだ個別要素がnumpy.int32であるため、df.equals的には一致しないようです。

個別要素をfor文で呼び出して要素ごとにintにして比較すれば出来るのでしょうが、こう、なんとかdf.equals等で一括で比較したいです。

どうすれば良いでしょうか？

　　　　import pandas as pd
　　　　 df = pd.DataFrame(columns=['number'])
　　　　#プログラム内データ
        d2 = pd.Series([int(number[0])], index=df.columns)
        df = df.append(d2,  ignore_index=True)
        #セーブデータとの比較
        df2 = pd.read_csv(r"c:\temp.csv", encoding='utf_8_sig', index_col=0, dtype=int)
        print(df2)
        print(df2.iloc[0, 0])
        print(type(df2.iloc[0, 0]))
        
        print(df)
        print(df.iloc[0, 0])
        print(type(df.iloc[0, 0]))

　　　　 print(df.equals(df2))

Answer 1

equalsはdtypeが同じじゃないとFalseになるようですね。
astypeでdtypeをそろえてから比較してみたらどうでしょう。

df2 = df2.astype(df.dtypes)
print(df.equals(df2))

Comments

[Aki1000]

ありがとうございます。動きました。