Python как извлечь все элементы списка, если они начинаются с одинакового набора символов - PullRequest
Новая
       16

Python как извлечь все элементы списка, если они начинаются с одинакового набора символов

0 голосов
/ 06 марта 2019

Я должен разбить список слов с тегами POS на подсписки согласно к тегу POS используется. Мой список выглядит так: 

List=[", -> ','", ". -> '!'", ". -> '.'", ". -> '?'", "CC -> 'but'", "CD -> 'hundred'",
      "CD -> 'one'", "DT -> 'the'", "EX -> 'There'","IN -> 'as'", "IN -> 'because'",
      "IN -> 'if'", "IN -> 'in'", "JJ -> 'Sure'", 'MD -> "\'ll"', "MD -> 'ca'",
      "MD -> 'can'", "MD -> 'will'", "MD -> 'would'", "NN -> 'Applause'",
      "NN -> 'anybody'", "NN -> 'doubt'", "NNP -> 'Syria'",
      "NNS -> 'Generals'", "NNS -> 'people'", "NNS -> 'states'",  "PRP -> 'it'",
      "PRP$ -> 'our'",  "RB -> 'there'", "RBR -> 'more'", "RP -> 'out'", "TO -> 'to'",
      "UH -> 'Oh'", "UH -> 'Wow'", "VB -> 'stop'", "VB -> 'want'", "VBD -> 'knew'",
      "VBD -> 'was'", "VBG -> 'allowing'", "VBG -> 'doing'", "VBG -> 'going'",
      "VBN -> 'called'", "VBP -> 'take'", 'VBZ -> "\'s"', "VBZ -> 'is'", 
      "WDT -> 'that'", "WP -> 'what'"]

Мой желаемый результат будет что-то вроде 

[["IN -> 'as'", "IN -> 'because'", "IN -> 'if'", "IN -> 'in'"],["UH -> 'Oh'", "UH -> 'Wow'"]]

или даже лучше 

CC = ['but']
CD = ['hundred', 'one']

Я искал довольно много, но единственная функция, которую я мог найти, которая хотя бы частично выполняет эту работу: 

from itertools import groupby
print([list(g) for k, g in groupby(List, key=lambda x: x[0])])

Я поиграл со значением x, но кажется, что norhing работает очень хорошо.

Я также подумал об использовании чего-то подобного: 

RB = []
for item in List:
    if item.startswith('RB'):
        g=re.findall('-> (.*)', item)
        RB.append(g)

Это, безусловно, должно сработать, но было бы больно делать это для примерно 40 различных POS-тегов. Должен быть более простой способ.

1 Ответ

0 голосов
/ 06 марта 2019

Использовать defaultdict : 

from collections import defaultdict

List = [", -> ','", ". -> '!'", ". -> '.'", ". -> '?'", "CC -> 'but'", "CD -> 'hundred'",
      "CD -> 'one'", "DT -> 'the'", "EX -> 'There'","IN -> 'as'", "IN -> 'because'",
      "IN -> 'if'", "IN -> 'in'", "JJ -> 'Sure'", 'MD -> "\'ll"', "MD -> 'ca'",
      "MD -> 'can'", "MD -> 'will'", "MD -> 'would'", "NN -> 'Applause'",
      "NN -> 'anybody'", "NN -> 'doubt'", "NNP -> 'Syria'",
      "NNS -> 'Generals'", "NNS -> 'people'", "NNS -> 'states'",  "PRP -> 'it'",
      "PRP$ -> 'our'",  "RB -> 'there'", "RBR -> 'more'", "RP -> 'out'", "TO -> 'to'",
      "UH -> 'Oh'", "UH -> 'Wow'", "VB -> 'stop'", "VB -> 'want'", "VBD -> 'knew'",
      "VBD -> 'was'", "VBG -> 'allowing'", "VBG -> 'doing'", "VBG -> 'going'",
      "VBN -> 'called'", "VBP -> 'take'", 'VBZ -> "\'s"', "VBZ -> 'is'", 
      "WDT -> 'that'", "WP -> 'what'"]

data = defaultdict(set)
for key, value in (_.split('->') for _ in List):
  d[key.strip()].add(value.strip().replace("'", '').replace('"', ''))
print(dict(data))

В результате: 

{',': {','}, '.': {'.', '!', '?'}, 'CC': {'but'}, 'CD': {'hundred','one'}, 'DT': {'the'}, 'EX': {'There'}, 'IN': {'in', 'because', 'as', 'if'}, 'JJ': {'Sure'}, 'MD': {'will', 'ca', 'll', 'can', 'would'}, 'NN': {'Applause', 'doubt', 'anybody'}, 'NNP': {'Syria'}, 'NNS': {'Generals', 'states', 'people'}, 'PRP': {'it'}, 'PRP$': {'our'}, 'RB': {'there'}, 'RBR': {'more'}, 'RP': {'out'}, 'TO': {'to'}, 'UH': {'Wow', 'Oh'}, 'VB': {'want', 'stop'}, 'VBD': {'was', 'knew'}, 'VBG': {'going', 'allowing', 'doing'}, 'VBN': {'called'}, 'VBP': {'take'}, 'VBZ': {'s', 'is'}, 'WDT': {'that'}, 'WP': {'what'}}
...