串联后pandas重新计算索引

Question

我有一个问题，我通过沿行轴串联（垂直堆叠）来生成熊猫数据框。

每个组成数据帧都有一个自动生成的索引（升序编号）。

串联后，我的索引被搞砸了：它的计数最多为n（其中n是相应数据帧的shape [0]），并在下一个数据帧从零重新开始。

我正在尝试“根据给定的当前顺序重新计算索引”或“重新索引”（或者我认为）。事实证明，这DataFrame.reindex似乎并没有在做。

这是我尝试做的事情：

train_df = pd.concat(train_class_df_list)
train_df = train_df.reindex(index=[i for i in range(train_df.shape[0])])

它失败并显示“无法从重复的轴重新索引”。我不想更改数据的顺序…只需要删除旧索引并设置一个保留行顺序的新索引即可。

Answer 1

垂直连接后，如果得到的索引为 [0，n）， 然后是 [0，m）
，则只需执行以下操作reset_index：

train_df.reset_index(drop=True)

（您可以使用进行就地操作inplace=True）。

import pandas as pd

>>> pd.concat([
    pd.DataFrame({'a': [1, 2]}), 
    pd.DataFrame({'a': [1, 2]})]).reset_index(drop=True)
    a
0   1
1   2
2   1
3   2