如何在Python中定义每个维度具有不同单位的二维数量数组

Question

我想定义一个包含发电机组燃料消耗数据的数组。X轴的单位是'kW‘，y轴的单位是'gal/hr’。有没有办法用Pint或在每个轴上有不同尺寸的类似库来创建Numpy兼容的数量数组？

Answer 1

我发现的最好的方法是使用实验性的pint-pandas库，并将数据存储为数据帧。

consumption_data = pd.DataFrame({
    "power": pd.Series([5, 10, 15, 20], dtype="pint[kW]"),
    "consumption": pd.Series([0.6, 0.9, 1.3, 1.6], dtype="pint[gal/hr]")})