如何从2D numpy数组中的每一行获取第一个元素？

Question

我有一个2D numpy数组。我正在寻找每行中特定元素首次出现的索引。输出将是一个（n x 2）数组，其中n是行数，每个条目都包含该特定元素第一次出现的x和y坐标。

非常感谢。

Answer 1

我不知道你的平均什么“S和Y坐标”，所以我假设你的意思行和列位置。

import numpy as np 
np.array([(s, list(row).index(your_element)) for s,row in enumerate(your_array)]) 

注意它会提高ValueError如果your_element是不包含在某行。

以下版本会给你一个可能包含比输入少的行的输出，但对于your_element从一行中缺失的情况不会引发ValueError。

np.array([(s, list(row).index(your_element)) for s,row in enumerate(your_array) if your_element in row]) 

Answer 2

如果我理解正确的问题，你想是这样的：

import numpy as N 
# 
nrows=5 
ncols=10 
# 
a=N.random.random((nrows,ncols)) 
b=-99*N.ones((nrows,2)) 
# 
for j in range(nrows): 
    for i in range(ncols): 
     if(a[j,i]<0.5): 
      b[j,0]=i 
      b[j,1]=j 
      continue 

Answer 3

>>> # generate some fake data: 
>>> A = NP.random.randint(5, 10, 100).reshape(10, 10) 
>>> A 
    array([[5, 7, 8, 8, 5, 6, 6, 9, 6, 9], 
     [9, 8, 8, 9, 5, 6, 6, 9, 8, 9], 
     [8, 5, 6, 7, 8, 9, 5, 8, 6, 7], 
     [5, 8, 8, 6, 9, 6, 8, 5, 8, 9], 
     [6, 9, 8, 8, 5, 7, 6, 5, 7, 6], 
     [7, 8, 6, 7, 6, 6, 7, 8, 6, 8], 
     [8, 6, 8, 9, 8, 8, 9, 6, 8, 7], 
     [8, 7, 8, 5, 9, 5, 7, 8, 6, 9], 
     [9, 6, 5, 9, 9, 8, 8, 9, 8, 8], 
     [6, 8, 5, 8, 6, 5, 8, 6, 8, 5]]) 

>>> # sort this 2D array along one axis (i chose row-wise) 
>>> A = NP.sort(A, axis=1) 
>>> A 
    array([[5, 5, 6, 6, 6, 7, 8, 8, 9, 9], 
     [5, 6, 6, 8, 8, 8, 9, 9, 9, 9], 
     [5, 5, 6, 6, 7, 7, 8, 8, 8, 9], 
     [5, 5, 6, 6, 8, 8, 8, 8, 9, 9], 
     [5, 5, 6, 6, 6, 7, 7, 8, 8, 9], 
     [6, 6, 6, 6, 7, 7, 7, 8, 8, 8], 
     [6, 6, 7, 8, 8, 8, 8, 8, 9, 9], 
     [5, 5, 6, 7, 7, 8, 8, 8, 9, 9], 
     [5, 6, 8, 8, 8, 8, 9, 9, 9, 9], 
     [5, 5, 5, 6, 6, 6, 8, 8, 8, 8]]) 

>>> # now diff the sorted array along the same axis 
>>> A1 = NP.diff(A ,axis=1) 

>>> # A1 contains non-zero values for "first occurrences" and 
>>> # zero values for repeat values 
>>> A1 
    array([[0, 1, 0, 0, 1, 1, 0, 1, 0], 
     [1, 0, 2, 0, 0, 1, 0, 0, 0], 
     [0, 1, 0, 1, 0, 1, 0, 0, 1], 
     [0, 1, 0, 2, 0, 0, 0, 1, 0], 
     [0, 1, 0, 0, 1, 0, 1, 0, 1], 
     [0, 0, 0, 1, 0, 0, 1, 0, 0], 
     [0, 1, 1, 0, 0, 0, 0, 1, 0], 
     [0, 1, 1, 0, 1, 0, 0, 1, 0], 
     [1, 2, 0, 0, 0, 1, 0, 0, 0], 
     [0, 0, 1, 0, 0, 2, 0, 0, 0]]) 

可以重新制定结果，A1，根据需要，例如，作为具有在至极相同的形状A1一个布尔阵列每个小区或者T/F根据该值是否在原始矩阵表示该值的第一次出现：

>>> ndx = A1==0 
>>> ndx 
    array([[ True, False, True, True, False, False, True, False, True], 
     [False, True, False, True, True, False, True, True, True], 
     [ True, False, True, False, True, False, True, True, False], 
     [ True, False, True, False, True, True, True, False, True], 
     [ True, False, True, True, False, True, False, True, False], 
     [ True, True, True, False, True, True, False, True, True], 
     [ True, False, False, True, True, True, True, False, True], 
     [ True, False, False, True, False, True, True, False, True], 
     [False, False, True, True, True, False, True, True, True], 
     [ True, True, False, True, True, False, True, True, True]], dtype=bool)