如何为python数据帧创建行计数器(枚举)?

时间:2018-10-07 08:38:57

标签: python pandas

我有一个pandas数据框,我想创建一个新的变量来对行进行计数。因此,在第一行中,我想要具有值1,在第二行中,我想要具有值2,依此类推。我无法使用索引,因为我的数据帧是训练/测试拆分的结果,因此其中的索引不会变为0、1、2,...,而是随机分布的。所以我真的必须以某种方式创建一个新变量。这个任务真的很容易解决,例如使用SQL,但是使用python我就迷路了。 有人可以帮我吗? 我已经在这里看到了类似问题的解决方案:Python enumeration/row counter in a pandas column。它是有帮助的,但前提是人们可以看到如何适应它。我看不到适应的方法。区别在于我必须枚举一个已经存在的数据框,因此正确的命令有点不同。

谢谢。

1 个答案:

答案 0 :(得分:2)

您可以使用numpy.arange

df['row_count'] = np.arange(len(df)) + 1