代码之家 › 专栏 › 技术社区 › levant pied

Numpy获取满足条件的坐标和坐标?

numpy python

levant pied · 技术社区 · 7 年前

例如,假设我有这个数组:

[[45 78 95 49 98 18 54]
 [36 77 92 15 68 25 89]
 [29 44 58 18 85 65 43]
 [19 63 62 78 48 12 43]
 [90 65 17 90 65 44 53]]

把它们归零

[[ 0 78  0 49  0  0  0]
 [36 77 92 15 68 25 89]
 [ 0 44  0 18  0  0  0]
 [19 63 62 78 48 12 43]
 [ 0 65  0 90  0  0  0]]

从结果数组中获取所有元素的坐标,这些元素为>50

 [[0 1]
  [1 1]
  [1 2]
  [1 4]
  [1 6]
  [3 1]
  [3 2]
  [3 3]
  [4 1]
  [4 3]]

比如说:

rs = numpy.array([0, 2, 4])[:, None]
cs = numpy.array([0, 2, 4, 5, 6])
a[rs, cs] = 0
res = numpy.argwhere(a > 50)

有没有更快的(因为我必须复制 a

1 回复 | 直到 7 年前

Paul Panzer 7 年前

下面是一个使用 ogrid

>>> i, j = np.ogrid[(*map(slice, a.shape),)]
>>> np.argwhere((a>50) & ((i|2==3) | (j|2==3)))
array([[0, 1],
       [1, 1],
       [1, 2],
       [1, 4],
       [1, 6],
       [3, 1],
       [3, 2],
       [3, 3],
       [4, 1],
       [4, 3]])

推荐文章

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

1 年前

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

1 年前

mchaudh4 · 用numpy表示三对角矩阵

1 年前

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

1 年前

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

1 年前

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

1 年前

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

1 年前

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

1 年前

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

1 年前

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

1 年前