从 Timedelta 和 NaN 数组中获取浮点数

Question

我有一个简单的问题，给我带来了很多麻烦：我有一个大的 2D 数组，它是datetime.Timedelta对象和的混合np.nan，简化如下：tdarray = np.array([dt.timedelta(days=5), np.nan])现在我想从 timedelta 对象中获取天数和浮点数/整数，同时保持np.nan原样，即结果应该是np.array([ 5., nan]).从 timedelta 对象中获取天数很容易.days，并应用数组应该使用的函数，例如，np.fromiter然后重新整形。但是如何捕获尝试从 NaN 获取天数时发生的错误？我尝试屏蔽，但这也失败了 AttributeErrorMaskedArray没有属性days。有什么简单的解决办法吗？

Answer 1

利用np.nan唯一不等于自身的对象这一事实。请注意，如果您的数组包含其他对象，则它们应该定义相等运算符，否则会引发错误。

tdarray = np.asarray([dt.timedelta(days=5), np.nan])

mask = tdarray == tdarray  # This gives array([True, False])

tdarray[mask] = [x.days for x in tdarray[mask]]

# Optionally cast to float

tdarray = tdarray.astype(np.float64)

或者你可以简单地重建阵列

tdarray = np.asarray([x.days if x == x else x for x in tdarray],

                     dtype=np.float64)

如果tdarray是 ND 数组 (N > 1) 那么

shape = tdarray.shape

tdarray = np.asarray([x.days if x == x else x 

                      for x in tdarray.ravel()],

                      dtype=np.float64).reshape(shape)