python - 有没有更简单的方法来更改 pandas 数据框的索引值？

Question

我正在获取一个数据框，将其分成两个数据框，然后我需要更改索引值，以便没有数字大于总行数。

代码如下:

dataset =   pd.read_csv("dataset.csv",usecols['row_id','x','y','time'],index_col=0)
splitvalue = math.floor((0.9)*786239)
train = dataset[dataset.time < splitvalue]
test = dataset[dataset.time >= splitvalue]

这是我正在做的改变。我想知道是否有更简单的方法:

test.index=range(test.shape[0])
test.index.rename('row_id',inplace=True)

有更好的方法吗？

Answer 1

尝试:

test = test.reset_index(drop=True).rename_axis('row_id')