我有两个numpy数组(2维)
a1 = array([["a","b"],["a","c"],["b","b"],["a","b"]])
a2 = array([["a","b"],["b","b"],["c","a"],["a","c"]])
得到这样一个矩阵最优雅的方法是什么:
array([[1,0,0,0],
[0,0,0,1],
[0,1,0,0],
[1,0,0,0]])
其中元素(i,j)为1,如果所有元素(a1[i,:]==a2[j,:]),否则为0
(涉及两个for循环的所有内容我认为都不优雅)
Tags:
如果您真的需要整数,请将其转换为
int
作为最后一步:相关问题 更多 >
编程相关推荐