loop in 2-dimension array in Python(list)

Question

I am searching one particular pattern and remove the File. I have written following code and it is working file but I feel I can reduce for loop when I am trying to remove File (except List comprehension)

rm_file_pat = ["*.abc*", "*.xyz"]
        rm_file_list = [ glob.glob(f_pat) for f_pat in rm_file_pat]
        for rm_file in rm_file_list:
            for _rm_file in rm_file:
                os.remove(_rm_file)

thefourtheye · Accepted Answer · 2013-11-13 05:39:54Z

1

You can flatten the rm_file_list using chain.from_iterable and then simply iterate over the list

import itertools
for rm_file in itertools.chain.from_iterable(rm_file_list):
    os.remove(rm_file)

edited Nov 13, 2013 at 5:39

answered Nov 13, 2013 at 5:34

thefourtheye

241k53 gold badges466 silver badges505 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

John La Rooy Over a year ago

Don't use map for a side-effect. The first version is better

thefourtheye Over a year ago

@gnibbler Even I was reluctant to add the map solution, but considering the fact that OP wants to remove for, I just added that.

John La Rooy · Accepted Answer · 2013-11-13 05:47:06Z

1

from glob import glob
rm_file_pat = ["*.abc*", "*.xyz"]
for rm_file in (fn for f_pat in rm_file_pat for fn in glob(f_pat))
    os.remove(rm_file)

answered Nov 13, 2013 at 5:47

John La Rooy

306k54 gold badges378 silver badges513 bronze badges

Collectives™ on Stack Overflow

loop in 2-dimension array in Python(list)

2 Answers 2

2 Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

2 Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related