在Pandas DataFrame中编号子序列

Question

我有一个由两列组成的读数 DataFrame： experiment和value 。 experiment键插入实验数据框； 一行中有500行具有相同的experiment而不同的value代表同一实验中的500个读数，其中DF中的顺序是获取数据的顺序。 然后500用于下一个实验，依此类推。

我想在实验中寻找基于时间的趋势，因此我假设我想在0-499中标记每个点pos ，然后再标记groupby('pos') 。 如何创建该pos列，一个递增的值，每次experiment重置时该值都会重置为0？ 我猜这与experiment恒定行数相同。

Answer 1

如果我理解正确的话...

>>> df = pd.DataFrame({'Experiment' : [1,1,1,2,2,2,2,3,3,3], 
                       'Value' : np.random.randn(10)})
>>> df

   Experiment     Value
0           1 -0.924851
1           1 -0.599875
2           1  0.069982
3           2 -1.106909
4           2  0.463922
5           2  0.210568
6           2 -0.171456
7           3 -0.768618
8           3 -0.269928
9           3  0.055613

您将在groupby之后使用cumcount()获得所需的效果：

>>> df['Position'] = df.groupby('Experiment').cumcount()
>>> df

   Experiment     Value  Position
0           1 -0.924851         0
1           1 -0.599875         1
2           1  0.069982         2
3           2 -1.106909         0
4           2  0.463922         1
5           2  0.210568         2
6           2 -0.171456         3
7           3 -0.768618         0
8           3 -0.269928         1
9           3  0.055613         2