代码之家 › 专栏 › 技术社区 › kmario23 Mazdak

计算NumPy数组上的“移动计数和”

numpy-ndarray counting numpy python

kmario23 Mazdak · 技术社区 · 7 年前

我有以下数组:

# input
In [77]: arr = np.array([23, 45, 23, 0, 12, 45, 45])

# result
In [78]: res = np.zeros_like(arr)

现在,我想计算一个并将其储存在 res 数组。

具体来说, 数组应为:

In [79]: res
Out[79]: array([1, 1, 2, 1, 1, 2, 3])

[1, 1, 2, 1, 1, 2, 3]

计数如果一个元素重新出现,直到我们到达数组的末尾。此元素特定的计数应作为结果返回。

我们应该如何使用NumPy内置函数来实现这一点?我试过用 numpy.bincount

1 回复 | 直到 7 年前

Paul Panzer 7 年前

不确定你会找到一个内置的,所以这里是一个自制使用argsort。

def running_count(arr):
    idx = arr.argsort(kind='mergesort')
    sarr = arr[idx]
    neq = np.where(sarr[1:] != sarr[:-1])[0] + 1
    run = np.ones(arr.shape, int)
    run[neq[0]] -= neq[0]
    run[neq[1:]] -= np.diff(neq)
    res = np.empty_like(run)
    res[idx] = run.cumsum()
    return res

例如:

>>> running_count(arr)
array([1, 1, 2, 1, 1, 2, 3])
>>> running_count(np.array(list("xabaaybeeetz")))
array([1, 1, 1, 2, 3, 1, 2, 1, 2, 3, 1, 1])

解释人:

推荐文章

Oxana Verkholyak · 为元组数组创建布尔掩码

8 年前

user3452134 · 如何连接Python阵列

8 年前

najmath · 如何在nparray中添加字符串

8 年前

j-dr · 重新排列视图为Ndaray

8 年前

Rafael Valero · 从ndarray Python 2.7中删除列和行

8 年前

G F · 如何基于numpy数组中的条件删除行?

8 年前

Alex Luya · 如何将元组列表转换为元组的numpy数组?

8 年前

Leo · Numpy sum()出现“keepdims”错误

8 年前

Xu Zhang · 在numpy中,如何将向量乘以数组/矩阵元素?

9 年前