Dot and Cross

Sort by

recency

|

358 Discussions

|

  • + 0 comments
    import numpy
    N = int(input())
    a = numpy.array([input().split() for _ in range(N)], int)
    b = numpy.array([input().split() for _ in range(N)], int)
    
    print(a @ b)
    
  • + 0 comments
    import numpy as np
    
    N = int(input())
    A1 = np.array([input().split() for _ in range(N)], int)
    A2 = np.array([input().split() for _ in range(N)], int)
    print(np.dot(A1, A2))
    
  • + 0 comments

    For Python3 Platform

    import numpy
    
    N = int(input())
    A = numpy.array([input().split() for _ in range(N)], int)
    B = numpy.array([input().split() for _ in range(N)], int)
    
    print(numpy.matmul(A, B))
    
  • + 0 comments
    import numpy as np
    N = int(input())
    A = [list(map(int,input().split())) for _ in range(N)]
    B = [list(map(int,input().split())) for _ in range(N)]
    all_ind_sum = [[] for _ in range(N)]
    for i in range(N):
        for j in range(N):
            ind_sum = 0
            for k in range(N):
                ind_sum += A[i][k] * B[k][j]
            all_ind_sum[i].append(ind_sum)
    print(np.array(all_ind_sum))
    
  • + 0 comments

    My compact solution…

    import sys
    import numpy as np
    
    print([
        (n := int(input())),
       (ab := np.loadtxt(sys.stdin, int))[:n] @ ab[n:]
    ][1])
    

    It's a one-liner, broken here for clarity.