python-3.x - 天体表 : set data type from another table

Question

我有一个 astropy.Table ，称之为 T :

  a      b     c      d
int32 float64 str1 float64
----- ------- ---- -------
    1     2.0    0    10.0
    4     5.0    1    20.0
    5     8.5    2    30.0

我想从一个 numpy 数组创建另一个表，但具有与 T 相同的列和数据类型

import numpy as np
A = np.array([
  [1, 2, 3, 4],
  [5, 6, 7, 8]])

我可以创建一个具有相同列名的表

S = Table(A, names=T.colnames)

但是如果我也尝试传递数据类型

S = Table(A, names=T.colnames, dtype=T.dtype)

然后我得到一个错误 ValueError: dtype must be a list or None , 和 list(T.dtype)刚刚返回 TypeError: 'numpy.dtype' object is not iterable如何将数据类型从一个表传递到另一个表？

Answer 1

您可以使用当前版本的 astropy 执行此操作:

Table(A, names=T.colnames, dtype=[T[name].dtype for name in T.colnames])

这已经在 astropy 的 master 分支中得到了改进，在 astropy 的下一个版本 4.2 中，您可以简单地执行以下操作:

Table(A, dtype=T.dtype)