假设我有一个numpy数组,如:[1,2,3,4,5,6]和另一个数组:[0,0,1,2,2,1]我想按组对第一个数组中的项求和(第二个数组)并获得n组结果的组号顺序(在这种情况下,结果将是[3,9,9]).我怎么做numpy?
我有两个相等长度1D numpy的阵列,id并且data,其中,id是重复的序列,命令对定义子窗口的整数data.例如,
id data
1 2
1 7
1 3
2 8
2 9
2 10
3 1
3 -10
Run Code Online (Sandbox Code Playgroud)
我想data通过分组id并采取最大值或最小值进行聚合.在SQL中,这将是一个典型的聚合查询SELECT MAX(data) FROM tablename GROUP BY id ORDER BY id.有没有办法可以避免Python循环并以矢量化方式执行此操作,或者我是否必须下载到C?