how to return output of a function as list

Question

def fetch_name(resume_text):
  tokenized_sentences = nltk.sent_tokenize(resume_text)
  for sentence in tokenized_sentences:
    for chunk in nltk.ne_chunk(nltk.pos_tag(nltk.word_tokenize(sentence), tagset='universal')):
      if hasattr(chunk, 'label'):# and chunk.label() == 'PERSON':
        chunk = chunk[0]
      (name, tag) = chunk
      if tag == 'NOUN':
          #print(name)
          #z.extend(name)
          return name             

path = '/home/python/resumes/*.txt'
files = glob.glob(path)
for n in files:
    with io.open(n,'r') as f:
        data=f.read()
        print(fetch_name(data))

Following is the output that i obtain from the code that follows. I need the output in a form of list ['Sharayu','neeraj'.....]

SHARAYU
Neeraj
Gregory
MEhul
Shashank
Sandeep
Atul
ADITYA
Manoj
SAGAR
Shreya

Instead of printing, append it into a list. – Nitin Pawar Oct 15 '18 at 12:17 — Nitin Pawar, Oct 15 '18 at 12:17

score 0 · Answer 1 · answered Oct 15 '18 at 12:18

0

Do it like this:

path = '/home/python/resumes/*.txt'
files = glob.glob(path)
data_list = []
for n in files:
    with io.open(n,'r') as f:
        data=f.read()
        data_list.append(fetch_name(data))
    print(data_list)

answered Oct 15 '18 at 12:18

Nitin Pawar

168
1
10

score 0 · Accepted Answer · answered Oct 15 '18 at 12:19

0

Using a list() would do that as below:

namelist=list()    
for n in files:
    with io.open(n,'r') as f:
        data=f.read()
        namelist.append(fetch_name(data))
print(namelist)

answered Oct 15 '18 at 12:19

Shivkumar Birnale

2,906
1
11
17

score 0 · Answer 3 · answered Oct 15 '18 at 12:20

If you want your function to return a list instead, you can initialize a list and append the names to the list:

def fetch_name(resume_text):
    names = []
    tokenized_sentences = nltk.sent_tokenize(resume_text)
    for sentence in tokenized_sentences:
        for chunk in nltk.ne_chunk(nltk.pos_tag(nltk.word_tokenize(sentence), tagset='universal')):
            if hasattr(chunk, 'label'):  # and chunk.label() == 'PERSON':
                chunk = chunk[0]
            (name, tag) = chunk
            if tag == 'NOUN':
                # print(name)
                # z.extend(name)
                names.append(name)
    return names

score 0 · Answer 4 · answered Oct 15 '18 at 12:23

0

You can use the append method to add an entry to a list.

resumes = []
for path in glob.glob('/home/python/resumes/*.txt'):
    with open(path, 'r', encoding='utf-8') as f:
        resumes.append(fetch_name(f.read()))
print(resumes)

answered Oct 15 '18 at 12:23

Jack Taylor

5,588
19
35

it returns me a list like this [u'SHARAYU', u'Neeraj', u'Gregory', u'MEhul', u'Shashank', u'Sandeep', u'Atul', u'ADITYA', u'Manoj', u'SAGAR', u'Shreya'] i want the list as ['SHARAYU', 'Neeraj'.....] – sharayu salunkhe Oct 15 '18 at 12:26
Ah, you're using Python 2. I'd recommend switching to Python 3 - then the `u` prefix will go away, as Python 3 uses Unicode by default. – Jack Taylor Oct 15 '18 at 12:32
can you suggest something that can work with python 2 – sharayu salunkhe Oct 15 '18 at 12:37
If you use Python 2, then using the Unicode strings like `[u'SHARAYU', u'Neeraj', ...]` is better practice than using byte strings like `['SHARAYU', 'Neeraj', ...]`. If you have any non-Ascii characters in any of your files, then using Unicode strings will make things easier. – Jack Taylor Oct 15 '18 at 12:46

how to return output of a function as list

4 Answers4