How to name a file in a for loop separately from loop iterations

Question

I have an array with size=(100,600,600) and want to split it into 4 (25,600,600) arrays and save them with names ldx1, ldx2, ldx3 and, ldx4 in n folders like below.

folder 1: ldx1, ldx2, ldx3 and, ldx4

folder 2: ldx1, ldx2, ldx3 and, ldx4

folder n: ldx1, ldx2, ldx3 and, ldx4

My code just saves the first file which is ldx1 in all folders. See my try below:


import numpy as np
import scipy.io

num_folders = 10
count=0

for i in range(num_folders):
   x=np.random.randint(0,1,size=(100,600,600))
   x_ = x[i:i+25,:]    
   count+=1
   scipy.io.savemat('fig/%d/ldx%d.mat' % (i,count),  mdict={'my_list':x_},do_compression=True)

Why don't you first reshape your array to (4, 25, 600, 600) then loop over that? — Barmar
– Barmar, Commented Feb 3, 2022 at 18:43

Tim Roberts · Accepted Answer · 2022-02-03 18:43:42Z

2

You'll need two loops:

import numpy as np
import scipy.io

num_folders = 10

for i in range(num_folders):
   x = np.random.randint(0,1,size=(100,600,600))
   for j in range(4):
       scipy.io.savemat(
           'fig/%d/ldx%d.mat' % (i,j),
           mdict={'my_list':x[j*25:j*25+25,:,:]},
           do_compression=True
        )

answered Feb 3, 2022 at 18:43

Tim Roberts

55.3k4 gold badges28 silver badges41 bronze badges

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

Collectives™ on Stack Overflow

How to name a file in a for loop separately from loop iterations

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related