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

用条件更改numpy数组中的每个值

numpy python

Eshaka · 技术社区 · 6 年前

我有一个二维数组,我从中得到了图像,现在它有0s和255s,我想把所有的255s都改成1s,这对于 for 循环。

for i in range(lenX):
    for j in range(lenY):
        if img[i,j]==255:
            img[i,j] = 1

这里img是我的阵列。我非常确定有一种更简单的方法可以使用某种麻木的函数或其他东西来完成它。但我找遍了找不到的地方。

如果你知道如何轻松做到这一点……请帮帮我

2 回复 | 直到 6 年前

Sociopath 6 年前

使用 np.where

import numpy as np 

a = np.array([[1,9,1],[12,15,255],[255,1,245],[23,255,255]]) 
a = np.where(a==255, 1, a)
print(a)

输出:

[[  1   9   1]                                                                                                                                                    
 [ 12  15   1]                                                                                                                                                    
 [  1   1 245]                                                                                                                                                    
 [ 23   1   1]]

Martin 6 年前

这样就可以在不带循环的条件下修改矩阵

img[img==255]=1

推荐文章

Redshoe · 当执行numpy.genfromtxt()时,python是否会读取文件的所有行?

4 月前

unfolx · numpy数组不等式的执行时间

5 月前

mchaudh4 · 用numpy表示三对角矩阵

5 月前

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

5 月前

LMC · Numpy数组布尔索引以获取包含元素

5 月前

HJA24 · 根据条件用值正向填充Numpy矩阵/掩码

6 月前

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

6 月前

Amarth Gûl · 找到一组向量的最近收敛点

6 月前

Mr. W · numpy.divide是函数、类还是其他什么?

6 月前

Mr. W · 为什么numpy.array在编辑内部数据时如此缓慢?

6 月前