如何将自定义函数应用于 xarray.Dataset 坐标的每个值？

Posted 2023-03-24

【中文标题】如何将自定义函数应用于 xarray.Dataset 坐标的每个值？

【英文标题】：How can I apply a custom function to every value of an xarray.Dataset coordinate?

【发布时间】：2022-01-17 02:44:23

【问题描述】：

我有一个带有“时间”坐标的 xarray.Dataset，type=datetime64。我想将其转换为 pd.Timestamp，以便我可以更轻松地将其他数据集与此数据集进行组合和比较。

有没有办法可以对时间坐标中的每个值执行 pd.Timestamp() 函数，或者有没有更简单的方法可以更改类型？谢谢！

【问题讨论】：

ds['time'] = pd.to_timestamp(ds.time.values) 有效吗？请参阅to_timestamp 文档。

【参考方案1】：

您可以使用 Python 中的 map() 函数将可迭代对象中的每个值应用于函数。

def addition(n):
    return n + n
  
numbers = (1, 2, 3, 4)
result = map(addition, numbers)
print(list(result))

Output >> [2, 4, 6, 8]