как вернуть вывод функции в виде списка

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))

Ниже приведен вывод, который я получаю из следующего кода.Мне нужен вывод в виде списка ['Sharayu', 'neeraj' .....]

SHARAYU
Neeraj
Gregory
MEhul
Shashank
Sandeep
Atul
ADITYA
Manoj
SAGAR
Shreya

Answer 1

Вы можете использовать метод добавления , чтобы добавить запись в список.

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)

Answer 2

Использование list() сделает это, как показано ниже:

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

Answer 3

Если вы хотите, чтобы ваша функция возвращала список, вы можете инициализировать список и добавить имена в список:

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

Answer 4

Сделай так:

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)