写一段代码让类别变量 “Vehicle Class” 的值转换为整数

在Python中，如果你有一个名为"Vehicle Class"的分类变量（例如字符串类型），你需要将其转换为整数形式，你可以创建一个映射字典（dictionary mapping）来进行这个操作。这里是一个简单的例子：

# 假设原始数据集中的车辆类别是字符串，比如{'Small': 1, 'Medium': 2, 'Large': 3}
class_mapping = {'Small': 1, 'Medium': 2, 'Large': 3}

def convert_to_int(vehicle_class):
    return class_mapping.get(vehicle_class, -1)  # 如果类别不在映射中，返回一个默认值如-1

# 测试数据
vehicle_classes = ['Small', 'Medium', 'Large', 'Unknown']
integer_classes = [convert_to_int(c) for c in vehicle_classes]
print(integer_classes)

这段代码首先定义了一个映射字典，然后定义了一个函数`convert_to_int`，通过`get`方法获取对应整数值。如果类别不存在于映射中，则返回默认值（在这里是-1）。最后对给定的车辆类别列表应用这个转换函数。