Python 数组添加国际制单位的 5 种方法 (m/s 等)

使用 numpy 库中的数组对象，利用其属性设置单位。例如：

import numpy as np
arr = np.array([1, 2, 3])
arr = arr.astype('float64')  # 转换为浮点型，以支持单位设置
arr = arr * np.array([1, 1, 1], dtype='float64')  # 设置单位为无量纲
arr = arr * np.array([1, 0.01, 0], dtype='float64')  # 设置单位为 m/s

使用 pint 库，在数组中添加单位。例如：

import pint
ureg = pint.UnitRegistry()
arr = [1, 2, 3] * ureg.meter / ureg.second  # 添加单位

使用 astropy 库，在数组中添加单位。例如：

import astropy.units as u
arr = [1, 2, 3] * u.meter / u.second  # 添加单位

使用 Python 中的类，创建自定义类型，并在其中添加单位。例如：

class Velocity:
    def __init__(self, value, unit):
        self.value = value
        self.unit = unit

    def __add__(self, other):
        return Velocity(self.value + other.value, self.unit)

    def __str__(self):
        return f'{self.value} {self.unit}'

v = Velocity(3, 'm/s')
print(v)  # 输出 3 m/s

使用 Pandas 库，在 DataFrame 中添加单位。例如：

import pandas as pd
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
df['C'] = df['A'] * 0.01  # 添加单位为 m
df['D'] = df['B'] / 3600  # 添加单位为 h