pymystem3 example

pymystem3_example.py

import sys, re, pymystem3

mystem = pymystem3.Mystem()

def stem(s):
  return [(e['text'].strip(), \
    e['analysis'][0]['lex'] \
       if 'analysis' in e and len(e['analysis']) > 0 else '', \
    re.match('^([A-Z]+)', e['analysis'][0]['gr']).group(0) \
       if 'analysis' in e and len(e['analysis']) > 0 else '', \
    ','.join(set(re.findall(r"[\w']+", e['analysis'][0]['gr'])[1:])) \
       if 'analysis' in e and len(e['analysis']) > 0 else '')\
    for e in mystem.analyze(s) if len(e['text'].strip()) > 0]