import pandas as pd

读取 CSV 文件

data = pd.read_csv('data.csv')

进行 One-Hot 编码

one_hot_columns = ['proto', 'service', 'state', 'attack_cat'] one_hot_data = pd.get_dummies(data[one_hot_columns])

将编码的值合并到原始数据中

data_encoded = pd.concat([data, one_hot_data], axis=1)

删除'id'和'label'列

data_encoded = data_encoded.drop(['id', 'label'], axis=1)

保存到新的 CSV 文件

data_encoded.to_csv('data_encoded.csv', index=False)

输出结果

print('编码后的数据已保存到 data_encoded.csv 文件')