R是Python中的seq_len等价物

时间:2018-05-27 04:09:05

标签: python r pandas refactoring

我正在尝试在data.frame中的组内进行索引

例如,在此数据框中

Col1
A
A
A
B
B
C
D
D
D

我想输出如下

Col1    idx
A   1
A   2
A   3
B   1
B   2
C   1
D   1
D   2
D   3

在R中,我可以使用data.table df[, idx:=seq_len(.N), by=Col1]执行以下操作。我在Python中找不到相应的东西。到目前为止,我知道我可以在linspace的软件包中使用arrangenumpy函数,但我不太确定如何按组进行操作。

提前谢谢。

1 个答案:

答案 0 :(得分:3)

使用 static int diagonalDifference(int[][] a, int n) { //I just edited this method int x=0, y=0,z,h=0; for(int i = 0; i < n; i++){ x = x + a[i][i]; } for(int j = n-1; j >= 0; j-- ){ y = y + a[h][j]; h++; } //System.out.println(x); //System.out.println(y); if(x>y){ z=x-y; }else{ z=y-x; } return(z); }

cumcount