我正在使用astropy.table模块中的Tables对象。
下面的代码显示了我正在处理的数据类型:
In [44]: table
Out[44]:
<Table length=9>
defocus source Chi2 xcentroid ycentroid FWHMx FWHMy Peak
float32 float32 float32 float32 float32 float32 float32 float32
------- ------- ------- --------- --------- ------- ------- -------
-0.3 0.0 346.648 2056.5 55.82 11.8635 11.8635 182.277
-0.3 4.0 148.302 2056.49 1911.02 6.66554 6.66554 299.074
-0.3 8.0 347.208 2056.51 3922.99 6.83129 6.83129 326.476
-0.26 0.0 318.489 2056.5 55.8803 10.206 10.206 195.055
-0.26 4.0 152.501 2056.51 1911.02 6.9012 6.9012 244.817
-0.26 8.0 285.845 2056.49 3922.99 7.7939 7.7939 236.194
-0.22 0.0 264.113 2056.5 55.9053 8.79704 8.79704 187.376
-0.22 4.0 163.228 2056.5 1911.02 2.43716 2.43716 402.182
-0.22 8.0 230.017 2056.5 3922.99 6.70312 6.70312 235.376
In [45]: type(table)
Out[45]: astropy.table.table.Table
In [46]: cols=table.columns
In [47]: type(cols)
Out[47]: astropy.table.table.TableColumns
In [48]: type(cols[0])
Out[48]: astropy.table.column.Column
In [50]: mylist_1 = [x for x in cols]
In [51]: mylist_2 = [cols[k] for k in range(len(cols))]
In [52]: type(mylist_1[0])
Out[52]: str
In [53]: type(mylist_2[0])
Out[53]: astropy.table.column.Column
In [54]: mylist_1[0]
Out[54]: 'defocus'
In [55]: mylist_2[0]
Out[55]:
<Column name='defocus' dtype='float32' length=9>
-0.3
-0.3
-0.3
-0.26
-0.26
-0.26
-0.22
-0.22
-0.22
我期待着这两行:
mylist_1 = [x for x in cols]
和
mylist_2 = [cols[k] for k in range(len(cols))]
将完全相同(第二个不那么优雅),但它没有,正如您在上面的输出中看到的那样(mylist_1仅包含列名,而不包含列本身)。为什么会这样?有什么我不太了解我的&#34; cols&#34;对象
感谢您的见解。
(我尝试将我的表写入fit文件时遇到了这个 - 使用astropy.io.fits - 这需要构建一个合适的列列表,这并不像人们期望的那样明显......)
答案 0 :(得分:2)
astropy.table.table.TableColumns
对象(通过columns
属性访问)是有序字典,而不是列表。这就是[x for x in cols]
给出列名列表的原因(与通过正常dict迭代给出键而不是值)相同。
为方便起见,它还支持索引访问和切片,如col[0]
或cols[1:3]
。但即使它支持类似列表的访问,TableColumns
首先是一个字典。
在PR #4394中,我们会添加一个新的itercols()
方法,该方法将支持cols_list = [x for x in t.itercols()]
。
答案 1 :(得分:0)
astropy.table.table.TableColumns是astropy包的已定义项。它是TableColumns类的一个实例,基本上你不应该使用迭代器“cols [k]”或“cols”来访问它。
但在课堂上,您可以定义方法包含()和 getitem ()来描述这两个请求的行为。见https://infohost.nmt.edu/tcc/help/pubs/python/web/special-methods.html
这里似乎定义的行为与经典的元素列表不相似。也许有一些doc可用于astropy类。
答案 2 :(得分:0)
你的循环在列上迭代略有不同(但等效)。 但你在列表理解中添加了不同的东西。列是dict
- 就像它们有一个索引和一个相关元素:
考虑这样一个表:
from astropy.table import Table
data_rows = [(1, 2.0, 'x'),
(4, 5.0, 'y'),
(5, 8.2, 'z')]
t = Table(rows=data_rows, names=('a', 'b', 'c'), meta={'name': 'first table'},
dtype=('i4', 'f8', 'S1'))
cols=t.columns
使用以下方法迭代列索引
[x for x in cols]
# ['a', 'b', 'c']
或
[k for k in range(len(cols))]
# [0, 1, 2]
这些似乎有所不同,但cols[0] == cols['a']
所以这些只是两种不同的方式来索引列。
但是如果你想迭代实际列而不是他们的索引你可以这样做:
[cols[x] for x in cols]
# [<Column name='a' dtype='int32' length=3>1 4 5,
# <Column name='b' dtype='float64' length=3>2.0 5.0 8.2,
# <Column name='c' dtype='bytes1' length=3> x y z]
或:
[cols[k] for k in range(len(cols))]
# [<Column name='a' dtype='int32' length=3>1 4 5,
# <Column name='b' dtype='float64' length=3>2.0 5.0 8.2,
# <Column name='c' dtype='bytes1' length=3> x y z]
这次我们实际插入索引列而不是插入索引。
但是,转换为FITS并不意味着您可以直接将表格保存为任何格式,请参阅their supported formats。
例如,写入拟合就像使用:
一样简单t.write('new_table.fits')
或
t.write('new_table.fits', format='fits')