将“类型”进行 one-hot 编码

可以使用以下代码将“类型”进行 one-hot 编码：

import numpy as np

# 假设有3种类型：A、B、C
types = ['A', 'B', 'C']

# 定义一个字典，将每种类型映射到一个整数
type_to_int = {t: i for i, t in enumerate(types)}

# 假设当前数据中的类型为 B
current_type = 'B'

# 创建一个全0的数组，长度为类型总数
one_hot = np.zeros(len(types))

# 将对应类型的位置设置为1
one_hot[type_to_int[current_type]] = 1

print(one_hot)

输出结果为：

[0. 1. 0.]

其中，`type_to_int` 是将每种类型映射到一个整数的字典，`current_type` 是当前数据中的类型，`one_hot` 是进行 one-hot 编码后的结果。