Python中优化条件定义多变量处理DataFrame列名差异的方法

1. 使用字典统一管理列名映射 #

把不同数据源的列名映射封装成字典，通过条件选择对应字典，相比零散变量更整洁，也方便后续扩展更多数据源类型：

# 定义不同数据源的列名映射字典
air_data_cols = {
    'lat': 'latitude',
    'lon': 'longitude',
    'alt': 'height_above_takeoff(feet)',
    'time': 'datetime(utc)',
    'head': 'compass_heading(degrees)',
    'pitch': 'gimbal_pitch(degrees)',
    'roll': 'roll(degrees)',
    'vid': 'isVideo'
}

other_data_cols = {
    'lat': 'OSD.latitude',
    'lon': 'OSD.longitude',
    'alt': 'OSD.height[ft]',
    'time': 'CUSTOM.updateTime[local]',
    'head': 'OSD.yaw',
    'pitch': 'OSD.pitch',
    'roll': 'OSD.roll',
    'vid': 'CAMERA.isVideo'
}

# 根据条件选择对应映射
LOG_COLS = air_data_cols if isAirData else other_data_cols

# 访问数据时直接用统一的键
lat_data = df[LOG_COLS['lat']]

2. 用数据类封装列名（带类型提示） #

如果需要更强的类型约束和代码可读性，可以用dataclasses模块封装列名配置，适合大型项目维护：

from dataclasses import dataclass

@dataclass
class DataFrameCols:
    lat: str
    lon: str
    alt: str
    time: str
    head: str
    pitch: str
    roll: str
    vid: str

# 实例化不同数据源的列名配置
air_data_config = DataFrameCols(
    lat='latitude',
    lon='longitude',
    alt='height_above_takeoff(feet)',
    time='datetime(utc)',
    head='compass_heading(degrees)',
    pitch='gimbal_pitch(degrees)',
    roll='roll(degrees)',
    vid='isVideo'
)

other_data_config = DataFrameCols(
    lat='OSD.latitude',
    lon='OSD.longitude',
    alt='OSD.height[ft]',
    time='CUSTOM.updateTime[local]',
    head='OSD.yaw',
    pitch='OSD.pitch',
    roll='OSD.roll',
    vid='CAMERA.isVideo'
)

# 选择配置
LOG_COLS = air_data_config if isAirData else other_data_config

# 访问示例
time_data = df[LOG_COLS.time]

3. 提前统一DataFrame列名（推荐） #

如果后续代码只需要统一的列名，可以在读取DataFrame后直接重命名列，彻底消除后续代码的分支判断：

# 定义重命名规则
rename_mapping = air_data_cols if isAirData else other_data_cols
# 反转映射：原列名 → 统一列名
reverse_rename = {v: k for k, v in rename_mapping.items()}

# 重命名DataFrame列
df = df.rename(columns=reverse_rename)

# 后续代码直接使用统一的列名，无需再判断数据源
lat_data = df['lat']
vid_status = df['vid']

这种方式让后续业务代码完全脱离对原始列名的依赖，逻辑更清晰，也减少了重复判断的冗余。

内容的提问来源于stack exchange，提问作者Ulises Bussi