python - Writing a 3D numpy array to files including its indices -


i'm new apologies have syntax right understand question!
have numpy array shape (2,2,2)

array([[[ 1, 1],         [ 2, 2]],         [[ 3, 3],         [ 4, 4]]]) 

how write file list indices , array value i.e.

0 0 0 1 1 0 0 3 0 1 0 2 1 1 0 4 0 0 1 1 1 0 1 3 0 1 1 2 1 1 1 4 

thanks, ingrid.

you can use numpy.ndenumerate

a = np.array([[[1, 1],                [2, 2]],                [[3, 3],                [4, 4]]])  items in np.ndenumerate(a):     print(items) 

output

((0, 0, 0), 1) ((0, 0, 1), 1) ((0, 1, 0), 2) ((0, 1, 1), 2) ((1, 0, 0), 3) ((1, 0, 1), 3) ((1, 1, 0), 4) ((1, 1, 1), 4) 

to remove parentheses can unpack everything

for indexes, value in np.ndenumerate(a):     x,y,z = indexes     print(x,y,z,value) 

output

0 0 0 1 0 0 1 1 0 1 0 2 0 1 1 2 1 0 0 3 1 0 1 3 1 1 0 4 1 1 1 4 

to handle file writing

with open('file.txt', 'w') f:     indexes, value in np.ndenumerate(a):         x,y,z = indexes         f.write('{} {} {} {}\n'.format(x,y,z,value)) 

Comments

Popular posts from this blog

javascript - IndexedDB error: Uncaught DataCloneError: Failed to execute 'put' on 'IDBObjectStore': An object could not be cloned -

amazon web services - When using Cognito credentials with AWS in a browser (javascript), keep getting "missing credentials" error -

c# - How to set the datatextfield value instead of the value on a drop down list? -