lsst.astshim g50ff169b8f+96c6868917
Loading...
Searching...
No Matches
fitsTableContinued.py
Go to the documentation of this file.
1__all__ = ["FitsTable", "getColumnData"]
2
3from ._astshimLib import DataType, FitsTable
4
5
6def getColumnData(self, column):
7 """Retrieve the column data in the correct type and shape.
8
9 Parameters
10 ----------
11 column : `str`
12 Name of the column to retrieve.
13
14 Returns
15 -------
16 data : `list` of `numpy.array`
17
18 """
19 nrows = self.nRow
20 shape = self.columnShape(column)
21 dtype = self.columnType(column)
22
23 if dtype == DataType.DoubleType:
24 newshape = list(shape)
25 newshape.append(nrows)
26 coldata = self.getColumnData1D(column)
27 coldata = coldata.reshape(newshape, order="F")
28 else:
29 raise ValueError("Can only retrieve double column data")
30 return coldata
31
32
33FitsTable.getColumnData = getColumnData