将onehot编码向量在紧凑的np.array中的位置返回

代码示例：

import numpy as np

def compact_onehot(onehot_vector):
    """
    将onehot编码向量在紧凑的np.array中的位置返回。
    """
    non_zeros = np.where(onehot_vector)[0]
    compact_positions = non_zeros - np.arange(len(non_zeros))
    return compact_positions

# 示例
onehot = np.array([0, 1, 0, 0, 0, 1])
compact_positions = compact_onehot(onehot)
print(compact_positions)
# 输出：[1 5]

该方法将onehot编码向量转换为一个只包含该向量非零元素位置的紧凑np.array。在这个紧凑np.array中，元素的位置即为该向量中原位置的索引值减去其前面的非零元素个数。