tf.setdiff1d

setdiff1d(x,y)也是一个比较奇怪,不常用的函数。
这个函数返回两个值:
第一个是:在x中出现,但是在y中没有出现的的元素
第二个是:这些元素在x中的索引(也就是下标或者位置)
我们举例说明:

import tensorflow as tf
import numpy as np
x=[2,3,1,4,2,5]
y=[2,5,6,7,8,3,9]
z,idx=tf.setdiff1d(x,y)
sess=tf.Session()
print("不同数据:\n",sess.run(z))
print("不同数据在x中的位置:\n",sess.run(idx))

输出结果如下:
不同数据:
[1 4]
不同数据在x中的位置:
[2 3

猜你喜欢

转载自blog.csdn.net/qq_31150463/article/details/84556029
今日推荐