Pandas：如何將增量值添加到新列？

Question

我有一個 dataframe ，我想在其中添加一個列 id，對於第一行，我想從一個精確的值開始（例如 start_id = 1200），並且為每一行添加 1 到前一個 id 的值。

dataframe：

    x   
1   paris   
2   paris  
3   lyon  
4   lyon   
5   toulouse

預期 output：

    id     x      
1   1200   paris   
2   1201   paris  
3   1202   lyon  
4   1203   lyon   
5   1204   toulouse

我試過這個但不起作用：

df.insert(start_id, 'id', range(start_id, 1 + len(df)))

我怎樣才能做到這一點？

Answer 1

一種使用range()和len(df)的方法，因此它將從1200開始並在 id column上遞增地添加row's

df['id'] = range(1200, 1200+len(df))

Answer 2

這里換一種方式：

df.assign(id=df.reset_index().index + 1200)

Output：

          x    id
1     paris  1200
2     paris  1201
3      lyon  1202
4      lyon  1203
5  toulouse  1204