Как я могу использовать Python itertools.groupby ()?
257670 просмотра
12 ответа
Я не смог найти понятного объяснения того, как на самом деле использовать itertools.groupby()
функцию Python . То, что я пытаюсь сделать, это:
- Возьмите список - в этом случае дети объективированного
lxml
элемента - Разделите его на группы по некоторым критериям
- Затем выполните итерацию по каждой из этих групп в отдельности.
Я просмотрел документацию и примеры , но у меня возникли проблемы при попытке применить их за пределы простого списка цифр.
Итак, как мне использовать itertools.groupby()
? Есть ли другая техника, которую я должен использовать? Указатели на хорошее "предварительное" чтение также приветствуются.
Ответы (12)
603 плюса
ВАЖНОЕ ПРИМЕЧАНИЕ: сначала нужно отсортировать данные .
Часть, которую я не получил, - это то, что в примере конструкции
groups = []
uniquekeys = []
for k, g in groupby(data, keyfunc):
groups.append(list(g)) # Store group iterator as a list
uniquekeys.append(k)
k
является текущим ключом группировки и g
является итератором, который можно использовать для перебора группы, определенной этим ключом группировки. Другими словами, сам groupby
итератор возвращает итераторы.
Вот пример этого, используя более ясные имена переменных:
from itertools import groupby
things = [("animal", "bear"), ("animal", "duck"), ("plant", "cactus"), ("vehicle", "speed boat"), ("vehicle", "school bus")]
for key, group in groupby(things, lambda x: x[0]):
for thing in group:
print "A %s is a %s." % (thing[1], key)
print " "
Это даст вам вывод:
Медведь это животное.
Утка это животное.Кактус это растение.
Скоростной катер - это транспортное средство.
Школьный автобус - это транспортное средство.
В этом примере things
это список кортежей, где первый элемент в каждом кортеже - это группа, к которой принадлежит второй элемент.
groupby()
Функция принимает два аргумента: (1) данные в группу и (2) функцию , чтобы сгруппировать его с.
Здесь lambda x: x[0]
указывается groupby()
использовать первый элемент в каждом кортеже в качестве ключа группировки.
В приведенном выше for
операторе groupby
возвращает три пары (ключ, групповой итератор) - один раз для каждого уникального ключа. Вы можете использовать возвращенный итератор для перебора каждого отдельного элемента в этой группе.
Вот немного другой пример с теми же данными, используя понимание списка:
for key, group in groupby(things, lambda x: x[0]):
listOfThings = " and ".join([thing[1] for thing in group])
print key + "s: " + listOfThings + "."
Это даст вам вывод:
Автор: James Sulak Размещён: 10.08.2008 06:45животные: медведь и утка.
растения: кактус.
транспортные средства: скоростной катер и школьный автобус.
73 плюса
Можете ли вы показать нам свой код?
Пример на документации по Python довольно прост:
groups = []
uniquekeys = []
for k, g in groupby(data, keyfunc):
groups.append(list(g)) # Store group iterator as a list
uniquekeys.append(k)
Итак, в вашем случае data - это список узлов, keyfunc - это то, куда идет логика функции вашего критерия, а затем groupby()
группирует данные.
Вы должны быть осторожны, чтобы отсортировать данные по критериям, прежде чем позвонить, groupby
иначе это не сработает. groupby
Метод на самом деле просто перебирает список и всякий раз, когда ключ меняется, он создает новую группу.
46 плюса
itertools.groupby
это инструмент для группировки предметов.
Из документов мы узнаем, что это может сделать:
# [k for k, g in groupby('AAAABBBCCDAABBB')] --> A B C D A B
# [list(g) for k, g in groupby('AAAABBBCCD')] --> AAAA BBB CC D
groupby
объекты дают пары ключей-групп, где группа является генератором.
Особенности
- А. Группируйте последовательные элементы вместе
- Б. Сгруппировать все вхождения элемента, учитывая отсортированный итеративный
- C. Укажите, как группировать элементы с помощью ключевой функции *
Сравнения
# Define a printer for comparing outputs
>>> def print_groupby(iterable, keyfunc=None):
... for k, g in it.groupby(iterable, keyfunc):
... print("key: '{}'--> group: {}".format(k, list(g)))
# Feature A: group consecutive occurrences
>>> print_groupby("BCAACACAADBBB")
key: 'B'--> group: ['B']
key: 'C'--> group: ['C']
key: 'A'--> group: ['A', 'A']
key: 'C'--> group: ['C']
key: 'A'--> group: ['A']
key: 'C'--> group: ['C']
key: 'A'--> group: ['A', 'A']
key: 'D'--> group: ['D']
key: 'B'--> group: ['B', 'B', 'B']
# Feature B: group all occurrences
>>> print_groupby(sorted("BCAACACAADBBB"))
key: 'A'--> group: ['A', 'A', 'A', 'A', 'A']
key: 'B'--> group: ['B', 'B', 'B', 'B']
key: 'C'--> group: ['C', 'C', 'C']
key: 'D'--> group: ['D']
# Feature C: group by a key function
>>> # keyfunc = lambda s: s.islower() # equivalent
>>> def keyfunc(s):
... """Return a True if a string is lowercase, else False."""
... return s.islower()
>>> print_groupby(sorted("bCAaCacAADBbB"), keyfunc)
key: 'False'--> group: ['A', 'A', 'A', 'B', 'B', 'C', 'C', 'D']
key: 'True'--> group: ['a', 'a', 'b', 'b', 'c']
Пользы
- Анаграммы ( см. Тетрадь )
- Binning
- Группа нечетных и четных чисел
- Группировать список по значениям
- Удалить дубликаты элементов
- Найти индексы повторяющихся элементов в массиве
- Разбить массив на куски размера n
- Найти соответствующие элементы между двумя списками
- Алгоритм сжатия ( см. Записную книжку ) / Кодировка длин серий
- Группировка букв по длине, функция клавиш ( см. Блокнот )
- Последовательные значения за порогом ( см. Записную книжку )
- Найти диапазоны чисел в списке или непрерывных элементов (см. Документы )
- Найти все связанные длинные последовательности
- Возьмите последовательные последовательности, которые удовлетворяют условию ( см. Связанный пост )
Примечание. Некоторые из последних примеров взяты из PyCon (выступления) Виктора Террона (на испанском языке) «Кунг-фу на рассвете с Itertools». Смотрите также groupby
исходный код, написанный на C.
* Функция, в которой все элементы передаются и сравниваются, влияя на результат. Другие объекты с основными функциями включают в себя sorted()
, max()
и min()
.
отклик
# OP: Yes, you can use `groupby`, e.g.
[do_something(list(g)) for _, g in groupby(lxml_elements, criteria_func)]
Автор: pylang
Размещён: 25.08.2017 02:26
39 плюса
Отличный трюк с groupby заключается в запуске кодирования длины в одну строку:
[(c,len(list(cgen))) for c,cgen in groupby(some_string)]
выдаст вам список из двух кортежей, где первый элемент - это символ, а второй - количество повторений.
Редактировать: Обратите внимание, что это то, что отличает семантику itertools.groupby
SQL GROUP BY
: itertools не (и вообще не может) заранее сортировать итератор, поэтому группы с одинаковым «ключом» не объединяются.
25 плюса
Другой пример:
for key, igroup in itertools.groupby(xrange(12), lambda x: x // 5):
print key, list(igroup)
результаты в
0 [0, 1, 2, 3, 4]
1 [5, 6, 7, 8, 9]
2 [10, 11]
Обратите внимание, что igroup является итератором (подитератором, как его называет документация).
Это полезно для разбиения генератора на части:
def chunker(items, chunk_size):
'''Group items in chunks of chunk_size'''
for _key, group in itertools.groupby(enumerate(items), lambda x: x[0] // chunk_size):
yield (g[1] for g in group)
with open('file.txt') as fobj:
for chunk in chunker(fobj):
process(chunk)
Еще один пример группового - когда ключи не отсортированы. В следующем примере элементы в xx сгруппированы по значениям в yy. В этом случае сначала выводится один набор нулей, затем набор единиц, а затем снова набор нулей.
xx = range(10)
yy = [0, 0, 0, 1, 1, 1, 0, 0, 0, 0]
for group in itertools.groupby(iter(xx), lambda x: yy[x]):
print group[0], list(group[1])
Производит:
0 [0, 1, 2]
1 [3, 4, 5]
0 [6, 7, 8, 9]
Автор: user650654
Размещён: 21.01.2013 04:54
21 плюса
ПРЕДУПРЕЖДЕНИЕ:
Список синтаксиса (groupby (...)) не будет работать так, как вы хотите. Кажется, он уничтожает внутренние объекты итератора, поэтому
for x in list(groupby(range(10))):
print(list(x[1]))
будет производить:
[]
[]
[]
[]
[]
[]
[]
[]
[]
[9]
Вместо списка (groupby (...)) попробуйте [(k, list (g)) для k, g в groupby (...)], или, если вы часто используете этот синтаксис,
def groupbylist(*args, **kwargs):
return [(k, list(g)) for k, g in groupby(*args, **kwargs)]
и получить доступ к функциональности groupby, избегая этих надоедливых (для маленьких данных) итераторов вместе.
Автор: RussellStewart Размещён: 16.11.2013 12:399 плюса
Я хотел бы привести еще один пример, где groupby без сортировки не работает. Адаптировано из примера Джеймса Сулака
from itertools import groupby
things = [("vehicle", "bear"), ("animal", "duck"), ("animal", "cactus"), ("vehicle", "speed boat"), ("vehicle", "school bus")]
for key, group in groupby(things, lambda x: x[0]):
for thing in group:
print "A %s is a %s." % (thing[1], key)
print " "
выход
A bear is a vehicle.
A duck is a animal.
A cactus is a animal.
A speed boat is a vehicle.
A school bus is a vehicle.
есть две группы с транспортным средством, тогда как можно ожидать только одну группу
Автор: kiriloff Размещён: 07.05.2013 08:097 плюса
@CaptSolo, я попробовал твой пример, но он не сработал.
from itertools import groupby
[(c,len(list(cs))) for c,cs in groupby('Pedro Manoel')]
Выход:
[('P', 1), ('e', 1), ('d', 1), ('r', 1), ('o', 1), (' ', 1), ('M', 1), ('a', 1), ('n', 1), ('o', 1), ('e', 1), ('l', 1)]
Как видите, есть два «О» и «2», но они попали в отдельные группы. Именно тогда я понял, что вам нужно отсортировать список, переданный функции groupby. Итак, правильное использование будет:
name = list('Pedro Manoel')
name.sort()
[(c,len(list(cs))) for c,cs in groupby(name)]
Выход:
[(' ', 1), ('M', 1), ('P', 1), ('a', 1), ('d', 1), ('e', 2), ('l', 1), ('n', 1), ('o', 2), ('r', 1)]
Просто помните, что если список не отсортирован, функция группировки не будет работать !
Автор: pedromanoel Размещён: 15.10.2009 03:416 плюса
Сортировка и групповой
from itertools import groupby
val = [{'name': 'satyajit', 'address': 'btm', 'pin': 560076},
{'name': 'Mukul', 'address': 'Silk board', 'pin': 560078},
{'name': 'Preetam', 'address': 'btm', 'pin': 560076}]
for pin, list_data in groupby(sorted(val, key=lambda k: k['pin']),lambda x: x['pin']):
... print pin
... for rec in list_data:
... print rec
...
o/p:
560076
{'name': 'satyajit', 'pin': 560076, 'address': 'btm'}
{'name': 'Preetam', 'pin': 560076, 'address': 'btm'}
560078
{'name': 'Mukul', 'pin': 560078, 'address': 'Silk board'}
Автор: Satyajit Das
Размещён: 01.08.2017 07:14
5 плюса
Как я могу использовать Python itertools.groupby ()?
Вы можете использовать groupby, чтобы группировать вещи для повторения. Вы даете groupby итерируемую и необязательную ключевую функцию / вызываемую функцию, с помощью которой проверяются элементы, когда они выходят из итерируемой, и она возвращает итератор, который дает двукратный набор результата вызываемой клавиши и фактических элементов в еще один повторяемый. Из справки:
groupby(iterable[, keyfunc]) -> create an iterator which returns
(key, sub-iterator) grouped by each value of key(value).
Вот пример группирования с использованием сопрограммы для группировки по счетчику, он использует вызываемый ключ (в данном случае coroutine.send
), чтобы просто подсчитать счетчик для сколь угодно большого количества итераций и сгруппированный суб-итератор элементов:
import itertools
def grouper(iterable, n):
def coroutine(n):
yield # queue up coroutine
for i in itertools.count():
for j in range(n):
yield i
groups = coroutine(n)
next(groups) # queue up coroutine
for c, objs in itertools.groupby(iterable, groups.send):
yield c, list(objs)
# or instead of materializing a list of objs, just:
# return itertools.groupby(iterable, groups.send)
list(grouper(range(10), 3))
печать
[(0, [0, 1, 2]), (1, [3, 4, 5]), (2, [6, 7, 8]), (3, [9])]
Автор: Aaron Hall
Размещён: 27.07.2015 06:06
1 плюс
Может пригодиться один полезный пример:
from itertools import groupby
#user input
myinput = input()
#creating empty list to store output
myoutput = []
for k,g in groupby(myinput):
myoutput.append((len(list(g)),int(k)))
print(*myoutput)
Пример ввода: 14445221
Пример выборки: (1,1) (3,4) (1,5) (2,2) (1,1)
Автор: Arko Размещён: 18.06.2017 05:161 плюс
Вы можете написать собственную групповую функцию:
def groupby(data):
kv = {}
for k,v in data:
if k not in kv:
kv[k]=[v]
else:
kv[k].append(v)
return kv
Run on ipython:
In [10]: data = [('a', 1), ('b',2),('a',2)]
In [11]: groupby(data)
Out[11]: {'a': [1, 2], 'b': [2]}
Автор: Sky
Размещён: 10.10.2018 05:53
Вопросы из категории :
- python Обработка XML в Python
- python Как я могу использовать Python itertools.groupby ()?
- python Python: На какой ОС я работаю?
- python Как я могу создать непосредственно исполняемое кроссплатформенное приложение с графическим интерфейсом на Python?
- python Вызов функции модуля с использованием его имени (строки)
- python Звук Питона («Колокол»)
- python Regex и unicode
- python Создать зашифрованный ZIP-файл в Python
- python Создайте базовый итератор Python
- python Функция транспонирования / распаковки (обратная сторона zip)?
- iteration Как эффективно перебирать каждую запись на карте Java?
- iteration Путь от рекурсии к итерации
- iteration Какой самый простой / лучший / самый правильный способ перебирать символы строки в Java?
- iteration Почему использование «for ... in» с итерацией массива - плохая идея?
- iteration Итерирование каждого символа в строке с использованием Python
- iteration Как мне перебрать каждый элемент в n-мерной матрице в MATLAB?
- iteration Как создать диапазон с шагом n в bash? (сгенерировать последовательность чисел с приращениями)
- iteration Как мне перебрать NSArray?
- iteration Итерация через ряд дат в Python