dedalus / dedalus / utils / concat_data.py

import numpy as np
import pylab as P
import h5py
import glob
import os

ddir = 'snap_00004'
field = '/fields/u/1'
cpus = glob.glob(os.path.join(ddir,'data.cpu*'))

files = []
dd = []
for cpu in cpus:
    files.append(h5py.File(cpu,'r'))
    dd.append(files[-1][field])
    
data = np.concatenate(dd,axis=0)
print data
for i in range(len(cpus)):
    files[i].close()

print data.real.min(), data.real.max()
data2 =np.fft.fftn(data)
P.imshow(data2[:,0,:].real)
P.colorbar()
P.show()
Tip: Filter by directory path e.g. /media app.js to search for public/media/app.js.
Tip: Use camelCasing e.g. ProjME to search for ProjectModifiedEvent.java.
Tip: Filter by extension type e.g. /repo .js to search for all .js files in the /repo directory.
Tip: Separate your search with spaces e.g. /ssh pom.xml to search for src/ssh/pom.xml.
Tip: Use ↑ and ↓ arrow keys to navigate and return to view the file.
Tip: You can also navigate files with Ctrl+j (next) and Ctrl+k (previous) and view the file with Ctrl+o.
Tip: You can also navigate files with Alt+j (next) and Alt+k (previous) and view the file with Alt+o.