pandas.Datafreamの重複したindexを書き換えたい

Question

下記のようなデータフレームがあります。

df1

	col1	col2
idx_A	9	3
idx_B	5	1
idx_C	6	4
idx_C	6	4
idx_D	10	2

idx_Cの重複があると、その後の計算に都合が悪いので、次のように書き換えて便宜的に違う行としたいです。

df2

	col1	col2
idx_A_0	9	3
idx_B_1	5	1
idx_C_2	6	4
idx_C_3	6	4
idx_D_4	10	2

次のようにしましたが、エラーが出てしまいました。

idx_list = ["idx_A_0", "idx_B_1", "idx_C_2", "idx_C_3", "idx_D_4"]
df2 = df1.set_axis(idx_list, axis=0)

ValueError: Length mismatch: Expected axis has 4 elements, new values have 5 elements

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

Answer 1

import pandas as pd
import io

csv_data = '''
,col1,col2
idx_A,9,3
idx_B,5,1
idx_C,6,4
idx_C,6,4
idx_D,10,2
'''
df = pd.read_csv(io.StringIO(csv_data), index_col=0)

#
df.index += [f'_{i}' for i in range(len(df))]
print(df)

	col1	col2
idx_A_0	9	3
idx_B_1	5	1
idx_C_2	6	4
idx_C_3	6	4
idx_D_4	10	2

Answer 2

以下のようにreset_indexした結果（連続する数値）を求めてset_indexすることでできます。

import pandas as pd
from io import StringIO

s = """,col1,col2
idx_A,9,3
idx_B,5,1
idx_C,6,4
idx_C,6,4
idx_D,10,2"""
df = pd.read_csv(StringIO(s), index_col=[0])

idx = df.reset_index()
df = df.set_index( df.index + '_' + idx.index.astype(str))
print(df)
"""
         col1  col2
idx_A_0     9     3
idx_B_1     5     1
idx_C_2     6     4
idx_C_3     6     4
idx_D_4    10     2
"""