mergeFiles.py 1.34 KB
Newer Older
Zhang Xin's avatar
Zhang Xin committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
import numpy as np
import os
from astropy.table import Table

multiFileDir = '/nfsdata/share/CSSOSDataProductsSims/C10_refCat/1/'


mergeKeys = ['parallax1',
 'interSpec_nuv',
 'interSpec_u',
 'interSpec_g',
 'interSpec_r',
 'interSpec_i',
 'interSpec_z',
 'interSpec_y',
 'interSpec_2MASS_H',
 'interSpec_2MASS_J',
 'interSpec_2MASS_Ks',
 'interSpec_GAIA_GAIA3.G',
 'interSpec_GAIA_GAIA3.Gbp',
 'interSpec_GAIA_GAIA3.Grp',
 'interSpec_GALEX_GALEX.FUV',
 'interSpec_GALEX_GALEX.NUV',
 'interSpec_LSST_u',
 'interSpec_LSST_g',
 'interSpec_LSST_r',
 'interSpec_LSST_i',
 'interSpec_LSST_z',
 'interSpec_LSST_y',
 'interSpec_PAN-STARRS_PS1.g',
 'interSpec_PAN-STARRS_PS1.r',
 'interSpec_PAN-STARRS_PS1.i',
 'interSpec_PAN-STARRS_PS1.z',
 'interSpec_PAN-STARRS_PS1.y',
 'interSpec_SLOAN_SDSS.u',
 'interSpec_SLOAN_SDSS.g',
 'interSpec_SLOAN_SDSS.r',
 'interSpec_SLOAN_SDSS.i',
 'interSpec_SLOAN_SDSS.z']
allFiles = os.listdir(multiFileDir)
mergeTable = None
outFn = ''
for fn in allFiles:
    if 'fits' not in fn:
        continue
    if mergeTable is None:
        mergeTable = Table.read(os.path.join(multiFileDir,fn))
        outFn = os.path.join(multiFileDir,fn[0:25]+'.fits')
    else:
        tTab = Table.read(os.path.join(multiFileDir,fn))
        for key in mergeKeys:
            mergeTable[key] = mergeTable[key] + tTab[key]

mergeTable.write(outFn,overwrite = True)