問(wèn)題描述

我有以下 numpy 數(shù)組:

I have the following numpy array:

import numpy as np
arr = np.array([[1,2,3,4,2000],
                [5,6,7,8,2000],
                [9,0,1,2,2001],
                [3,4,5,6,2001],
                [7,8,9,0,2002],
                [1,2,3,4,2002],
                [5,6,7,8,2003],
                [9,0,1,2,2003]
              ])

我理解 np.sum(arr, axis=0) 提供結(jié)果:

array([   40,    28,    36,    34, 16012])

我想做的(沒(méi)有for循環(huán))是根據(jù)最后一列的值對(duì)列求和，以便提供的結(jié)果是:

what I would like to do (without a for loop) is sum the columns based on the value of the last column so that the result provided is:

array([[   6,    8,   10,   12, 4000],
       [  12,    4,    6,    8, 4002],
       [   8,   10,   12,    4, 4004],
       [  14,    6,    8,   10, 4006]])

我意識(shí)到如果沒(méi)有循環(huán)可能會(huì)有些牽強(qiáng)，但希望能做到最好……

I realize that it may be a stretch to do without a loop, but hoping for the best...

如果必須使用 for 循環(huán)，那將如何工作?

If a for loop must be used, then how would that work?

我試過(guò) np.sum(arr[:, 4]==2000, axis=0) (我會(huì)用 for 循環(huán)中的變量替換 2000)，但是它給出了 2

I tried np.sum(arr[:, 4]==2000, axis=0) (where I would substitute 2000 with the variable from the for loop), however it gave a result of 2

推薦答案

你可以在純 numpy 中使用 np.diff 和 np.add.reduceat.np.diff 將為您提供最右側(cè)列更改的索引:

You can do this in pure numpy using a clever application of np.diff and np.add.reduceat. np.diff will give you the indices where the rightmost column changes:

d = np.diff(arr[:, -1])

np.where 會(huì)將您的布爾索引 d 轉(zhuǎn)換為 np.add.reduceat 期望的整數(shù)索引:

np.where will convert your boolean index d into the integer indices that np.add.reduceat expects:

d = np.where(d)[0]

reduceat 也期望看到零索引，并且所有內(nèi)容都需要移動(dòng)一:

reduceat will also expect to see a zero index, and everything needs to be shifted by one:

indices = np.r_[0, e + 1]

使用 np.r_ 這里比 方便一點(diǎn)np.concatenate 因?yàn)樗试S標(biāo)量.然后總和變?yōu)?

Using np.r_ here is a bit more convenient than np.concatenate because it allows scalars. The sum then becomes:

result = np.add.reduceat(arr, indices, axis=0)

這當(dāng)然可以組合成一條線:

This can be combined into a one-liner of course:

>>> result = np.add.reduceat(arr, np.r_[0, np.where(np.diff(arr[:, -1]))[0] + 1], axis=0)
>>> result
array([[   6,    8,   10,   12, 4000],
       [  12,    4,    6,    8, 4002],
       [   8,   10,   12,    4, 4004],
       [  14,    6,    8,   10, 4006]])

這篇關(guān)于Numpy:條件總和的文章就介紹到這了，希望我們推薦的答案對(duì)大家有所幫助，也希望大家多多支持html5模板網(wǎng)！