Python Numpy。用2个矩阵进行操作

时间:2018-03-22 15:20:58

标签: python python-3.x numpy matrix

我有两个大小相同的CSV文件。值为10 s。

我需要循环2个文件(矩阵)并使用以下逻辑创建一个新矩阵:

if matrix A value = 1 and matrix B value = 1 
  then 
      result value is 0, 

if 1 and 0 
  then 
      0, 

if 0 and 0 
  then 
      0.

A = [
    [1, 0, 1],
    [1, 1, 1]
]
B = [
    [1, 0, 0],
    [1, 0, 0]
]
=>
C = [
    [0, 0, 1],
    [0, 1, 1]
]

我知道Numpy用于循环和操作矩阵和数组,但我坚持找到如何以适当的方式做到这一点。

1 个答案:

答案 0 :(得分:2)

这是获得所需输出的一种方法,但我认为你描述的逻辑并不完全是你的意思。这会输出一个1数组,其中您的矩阵彼此不同,0它们是相似的。

A = np.array([
    [1, 0, 1],
    [1, 1, 1]
])

B = np.array([
    [1, 0, 0],
    [1, 0, 0]])


C = (A != B).astype('int')

array([[0, 0, 1],
       [0, 1, 1]])