简单的问题,我想获得1D numpy数组.
给定一个2D数组,其中每行包含一个'1'值,如何将其转换为一维数组,由二维数组中的'1'的列索引组成
[[ 0. 0. 1.]
[ 0. 0. 1.]
[ 0. 0. 1.]
[ 0. 1. 0.]
[ 0. 1. 0.]
[ 0. 1. 0.]
[ 0. 1. 0.]
[ 1. 0. 0.]]
至
[2 2 1 1 1 1 0]
我怎么能在python中做到这一点？我不知道它的术语,让我知道这种转换的正确术语.