代码之家 › 专栏 › 技术社区 › alpelito7

如何只获取numpy数组中每个递增值的第一次出现?

numpy arrays python

alpelito7 · 技术社区 · 5 月前

在研究首次通过概率时,我遇到了这个问题。我想找到一种NumPythonic方法(没有显式循环),在每一行中只留下第一个严格递增的值 numpy 数组,同时将重复或非递增的值替换为零。例如,如果

arr = np.array([
    [1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 5, 5],
    [1, 1, 2, 2, 2, 3, 2, 2, 3, 3, 3, 4, 4],
    [3, 2, 1, 2, 1, 1, 2, 3, 4, 5, 4, 3, 2]])

我想得到输出:

out = np.array([
    [1, 0, 0, 2, 0, 0, 3, 0, 0, 4, 0, 5, 0],
    [1, 0, 2, 0, 0, 3, 0, 0, 0, 0, 0, 4, 0],
    [3, 0, 0, 0, 0, 0, 0, 0, 4, 5, 0, 0, 0]])

2 回复 | 直到 5 月前

wim 5 月前

每行可以累积的最大值:

>>> arr
array([[1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 5, 5],
       [1, 1, 2, 2, 2, 3, 2, 2, 3, 3, 3, 4, 4],
       [3, 2, 1, 2, 1, 1, 2, 3, 4, 5, 4, 3, 2]])
>>> np.maximum.accumulate(arr, axis=1)
array([[1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 5, 5],
       [1, 1, 2, 2, 2, 3, 3, 3, 3, 3, 3, 4, 4],
       [3, 3, 3, 3, 3, 3, 3, 3, 4, 5, 5, 5, 5]])

然后,您可以轻松屏蔽非递增值:

>>> m_arr = np.maximum.accumulate(arr, axis=1)
>>> np.where(np.diff(m_arr, axis=1, prepend=0), arr, 0)
array([[1, 0, 0, 2, 0, 0, 3, 0, 0, 4, 0, 5, 0],
       [1, 0, 2, 0, 0, 3, 0, 0, 0, 0, 0, 4, 0],
       [3, 0, 0, 0, 0, 0, 0, 0, 4, 5, 0, 0, 0]])

ouroboros1 5 月前

这里有一种方法:

m = np.hstack(
    (np.ones((arr.shape[0], 1), dtype=bool),
     np.diff(np.fmax.accumulate(arr, axis=1)) >= 1)
     )

out = np.zeros_like(arr)

out[m] = arr[m]

输出:

array([[1, 0, 0, 2, 0, 0, 3, 0, 0, 4, 0, 5, 0],
       [1, 0, 2, 0, 0, 3, 0, 0, 0, 0, 0, 4, 0],
       [3, 0, 0, 0, 0, 0, 0, 0, 4, 5, 0, 0, 0]])

解释

使用 np.fmax + np.ufunc.accumulate 以获得每行的运行最大值。
现在,检查一下 np.diff 大于或等于1。
使用 np.hstack 在列前添加 True 第一列(通过 np.ones ).
最后,初始化一个具有相同形状的零数组 arr (通过 np.zeros_like )并为掩码设置值。

推荐文章

Megrez7 · C#ToArray转换合并为一行,导致数组元素更改

3 月前

bairog · 从按属性筛选的对象数组字典中创建值数组

3 月前

Anka HanÄ±m · 关于结构和动态数组地址的问题

3 月前

Swapnil Supekar · 将二维阵列转换为一维阵列,并将一维阵列粘贴到另一张图纸上

4 月前

Lorenzo Bertolaccini · 在Angular项目中通过对话框后,在控制台中显示但在HTML中不显示的数据数组

4 月前

MaSc. H. · 大小与阵列大小

4 月前

Geremia · 2D NumPy数组+1D数组?

4 月前

MARTIN · 交换第一个和最后一个单词,反转所有中间的字符

5 月前

Paul Williams · 迭代数组时输出有问题

5 月前

Oliver Morgan · PostgreSQL错误:无法累积不同维度的数组

5 月前