Python: how to find the element in a list which match part of the name of the element

Question

I have a list of keyword to find from a list of file name. Example, if the keyword is 'U12', I want to find the csv file that contain 'U12' which is 'h_ABC_U12.csv'and print it out.

word = ['U12','U13','U14']
file_list = ['h_ABC_U12.csv','h_GGG_U13.csv','h_HVD_U14.csv','h_MMMB_U15.csv']

for x in range (len(word)):
    if word[x] in file_list:
       #print the file name

This is part of the code but unable to continue after some searches. I need the full name that match the word to print out.

score 1 · Answer 1 · answered May 23 '17 at 14:14

1

Try this -

for w in word:
     for file in file_list:
         if w in file:
             print file

answered May 23 '17 at 14:14

kuro

3,214
3
15
31

score 1 · Answer 2 · answered May 23 '17 at 14:19

1

You can try this using list comprehension:

word = ['U12','U13','U14']
file_list =['h_ABC_U12.csv','h_GGG_U13.csv','h_HVD_U14.csv','h_MMMB_U15.csv']

print [i for i in file_list for b in word if b in i]

answered May 23 '17 at 14:19

Ajax1234

69,937
8
61
102

score 1 · Accepted Answer · answered May 23 '17 at 14:21

I suggest using os.path.splitext to get the filename without extension.

>>> from os.path import splitext
>>> for f in file_list:
...     name = splitext(f)[0]
...     if any(name.endswith(tail) for tail in word):
...         print(name)
... 
h_ABC_U12
h_GGG_U13
h_HVD_U14

score 0 · Answer 4 · answered May 23 '17 at 14:12

0

Surely

for x in range(len(word)):
    for file in file_list:
        if x in file:
            print file

Would do the job?

answered May 23 '17 at 14:12

Henry

1,646
12
28

score 0 · Answer 5 · answered May 23 '17 at 14:19

This should perform better:

for file_name in file_list:
    the_word = next((w for w in word if w in file_name), None)
    if the_word:
        print the_word
        print file_name

Or to get the list of all file names:

[file_name for file_name in file_list if next((w for w in word if w in file_name), None)]

score 0 · Answer 6 · answered May 23 '17 at 15:09

word = ['U12','U13','U14']
file_list = ['h_ABC_U12.csv','h_GGG_U13.csv','h_HVD_U14.csv','h_MMMB_U15.csv']

for file_name in file_list:
  for w in word:
    if w in file_name:
      print(w,file_name)

RESULT

U12 h_ABC_U12.csv
U13 h_GGG_U13.csv
U14 h_HVD_U14.csv

Python: how to find the element in a list which match part of the name of the element

6 Answers6

Linked