python - 创建字符串数组

Question

我有一组用于各种类型的标志:

Data Type1 Type2 Type3
12   1     0     0
14   0     1     0
3    0     1     0
45   0     0     1

我想创建以下数组:

Data TypeName
12   Type1   
14   Type2   
3    Type2   
45   Type3   

我尝试创建一个空的字符串类型数组:

import numpy as np
z1 = np.empty(4, np.string_)
z1[np.where(Type1=1)] = 'Type1'

但这似乎并没有给我想要的结果。

编辑:我可以使用 pandas 数据框，每行只有 1 种类型，Type1、Type2、Type3

编辑2:Data Type1 Type2 Type3 是 pandas dataframe 中的列名，但我使用的是带有隐式名称的 numpy 数组，正如我在上面的示例中指出的那样。

Answer 1

这是一种滥用的方法，即我们从 Type1 列开始的每行只有一个 1 和 idxmax() 获取每行唯一出现的它 -

pd.concat((df.Data, df.iloc[:,1:].idxmax(1)),axis=1)

sample 运行-

In [42]: df
Out[42]: 
   Data  Type1  Type2  Type3
0    12      1      0      0
1    14      0      1      0
2     3      0      1      0
3    45      0      0      1

In [43]: pd.concat((df.Data, df.iloc[:,1:].idxmax(1)),axis=1)
Out[43]: 
   Data      0
0    12  Type1
1    14  Type2
2     3  Type2
3    45  Type3