удалить первый элемент в подвид матрицы

Question

У меня есть набор данных, как это:

[[0,1], 
[0,2], 
[0,3], 
[0,4], 
[1,5], 
[1,6], 
[1,7], 
[2,8], 
[2,9]] 

нужно удалить первые элементы каждого подвид данных, как определено в первом столбце. Поэтому сначала я получаю все элементы, которые имеют 0 в первом столбце, и удаляют первую строку: [0,1]. Затем я получаю элементы с 1 в первом столбце и удаляю первую строку [1,5], следующий шаг я удаляю [2,8] и т. Д. И т. Д. В конце концов, я хотел бы иметь набор данных, как это:

[[0,2], 
[0,3], 
[0,4], 
[1,6], 
[1,7], 
[2,9]] 

---

EDIT: Может ли это быть сделано в NumPy? Мой набор данных очень велик, поэтому для циклов на всех элементах требуется не менее 4 минут.

Answer 1

В соответствии с просьбой, numpy решения:

import numpy as np 
a = np.array([[0,1], [0,2], [0,3], [0,4], [1,5], [1,6], [1,7], [2,8], [2,9]]) 
_,i = np.unique(a[:,0], return_index=True) 

b = np.delete(a, i, axis=0) 

---

(выше отредактировано, чтобы включить @ решения Хайма, вот мое первоначальная маскировка решение ради потомства)

m = np.ones(len(a), dtype=bool) 
m[i] = False 
b = a[m] 

---

Intere stingly, маска, кажется, быстрее:

In [225]: def rem_del(a): 
    .....:  _,i = np.unique(a[:,0], return_index=True) 
    .....:  return np.delete(a, i, axis = 0) 
    .....: 

In [226]: def rem_mask(a): 
    .....:  _,i = np.unique(a[:,0], return_index=True) 
    .....:  m = np.ones(len(a), dtype=bool) 
    .....:  m[i] = False 
    .....:  return a[m] 
    .....: 

In [227]: timeit rem_del(a) 
10000 loops, best of 3: 181 us per loop 

In [228]: timeit rem_mask(a) 
10000 loops, best of 3: 59 us per loop 

Answer 2

Вы хотите использовать itertools.groupby() с тиром itertools.islice() и itertools.chain:

from itertools import islice, chain, groupby 
from operator import itemgetter 

list(chain.from_iterable(islice(group, 1, None) 
         for key, group in groupby(inputlist, key=itemgetter(0)))) 

• В groupby() группы вызова списка ввода в куски, где первый элемент является одинаковым (itemgetter(0) является ключевой группировкой).
• Звонок islice(group, 1, None) вызывает группы в итерации, где элемент сначала будет пропущен.
• Вызов chain.from_iterable() вызывает каждый результат islice() и объединяет их в новый итерируемый, который list() возвращается в список.

Демо:

>>> list(chain.from_iterable(islice(group, 1, None) for key, group in groupby(inputlist, key=itemgetter(0)))) 
[[0, 2], [0, 3], [0, 4], [1, 6], [1, 7], [2, 9]] 

Answer 3

a = [[0,1], 
[0,2], 
[0,3], 
[0,4], 
[1,5], 
[1,6], 
[1,7], 
[2,8], 
[2,9]] 

a = [y for x in itertools.groupby(a, lambda x: x[0]) for y in list(x[1])[1:]] 

print a 

Answer 4

Мой ответ:

from operator import itemgetter 
sorted(l, key=itemgetter(1)) # fist sort by fist element of inner list 
nl = [] 
[[0, 1], [0, 2], [0, 3], [0, 4], [1, 5], [1, 6], [1, 7], [2, 8], [2, 9]] 
j = 0; 
for i in range(len(l)): 
    if(j == l[i][0]): 
     j = j + 1 # skip element 
    else: 
     nl.append(l[i]) # otherwise append in new list 

выход:

>>> nl 
[[0, 2], [0, 3], [0, 4], [1, 6], [1, 7], [2, 9]] 

Answer 5

Pass в списках и ключ, который вы хотитепроверьте значения.

def getsubset(set, index): 
    hash = {} 
    for list in set: 
     if not list[index] in hash: 
      set.remove(list) 
      hash[list[index]] = list 

    return set