张量中第二个重复元素的遮罩

时间:2020-07-30 13:39:49

标签: python tensorflow machine-learning

给出一个整数张量,如何生成第二个重复元素的掩码
例如:

张量-> [13,2,31,46,56,2,53,46 ]

我要输出

[0,0,0,0,0,1,0,1]

张量流中的代码实现

1 个答案:

答案 0 :(得分:0)

尝试:

t = [13,2,31,46,56,2,53,46]
list(pd.Series(t).duplicated().astype('int'))

结果:

[0, 0, 0, 0, 0, 1, 0, 1]