python - 使用字典/表的第一列作为索引/键

Question

我正在尝试将 astropy.table 用作二维数组，但我也想将第一列用作索引。

例如，我已经有如下表格:

 a   b  c  d
'1t'  2  3  4
'5t'  6  7  8
'9t'  0  10 11

这样打印表['b'] = 2,6,0

我想要的是选择一个元素，这样打印表['b']['5t'] = 6

除了使用 table['b'][1] = 6 之外，还有其他方法吗？

Answer 1

我用这个例子展示了一个替代解决方案:

from astropy.table import Table, Column
import numpy as np
t = Table([['4t', '5t', '6t'], [0.1, 0.2, 0.3],[10, 20, 30]], names=('a', 'b','c'))
print (t)

输出:

 a   b   c 
--- --- ---
 4t 0.1  10
 5t 0.2  20
 6t 0.3  30

设置索引并使用loc:

t.add_index('a')
T=t.loc
print (T['4t']['b'])

输出:

0.1