numpy数组和one-hot编码的转换

import numpy as np
from keras.utils import to_categorical

# numpy 转换为 one-hot编码
data01 = np.array([0,1,3,2,5,6,4,7,9,8]) # 10个连续不重复数字（0-9）
data01_one_hots = to_categorical(data01)
print(data01_one_hots)
print(data01_one_hots.shape) # (10, 10)
print("-----------------------------------------------------")
data02 = np.array([0,1,3,2,5,6,4,7,9,8,10,11]) # 12个连续不重复数字（0-11）
data02_one_hots = to_categorical(data02)
print(data02_one_hots)
print(data02_one_hots.shape) # (12, 12)
print("-----------------------------------------------------")
data03 = np.array([0,1,3,2,5,6,4,7,9,8,2,3]) # 12个连续有重复数字（0-9）
data03_one_hots = to_categorical(data03)
print(data03_one_hots)
print(data03_one_hots.shape) # (12, 10)
print("-----------------------------------------------------")
data04 = np.array([0,1,11,2,5,6,4,7,14,8]) # 10个不连续的数字（0-14）
data04_one_hots = to_categorical(data04)
print(data04_one_hots)
print(data04_one_hots.shape) # (10, 15)

# one 转化为 numpy数组编码
data01 = [np.argmax(one_hot) for one_hot in data01_one_hots]
print(data01) # [0, 1, 3, 2, 5, 6, 4, 7, 9, 8]
data02 = [np.argmax(one_hot) for one_hot in data02_one_hots]
print(data02) # [0, 1, 3, 2, 5, 6, 4, 7, 9, 8, 10, 11]
data03 = [np.argmax(one_hot) for one_hot in data03_one_hots]
print(data03) # [0, 1, 3, 2, 5, 6, 4, 7, 9, 8, 2, 3]
data04 = [np.argmax(one_hot) for one_hot in data04_one_hots]
print(data04) # [0, 1, 11, 2, 5, 6, 4, 7, 14, 8]