如何将二维Numpy数组转换为一个热编码？

import numpy as np
from sklearn.preprocessing import OneHotEncoder

# 创建一个二维Numpy数组
data = np.array([[1, 2, 3],
                 [2, 3, 4],
                 [3, 4, 5]])

# 创建OneHotEncoder对象
encoder = OneHotEncoder(sparse=False)

# 对数据进行热编码
encoded_data = encoder.fit_transform(data)

# 打印热编码后的结果
print(encoded_data)

[[1. 0. 0. 1. 0. 0. 1. 0. 0.]
 [0. 1. 0. 0. 1. 0. 0. 1. 0.]
 [0. 0. 1. 0. 0. 1. 0. 0. 1.]]