使用NumPy替换Python语言中的嵌套循环

import numpy as np

# 使用嵌套循环计算两个数组的乘积
def multiply_arrays(a, b):
    result = []
    for i in range(len(a)):
        row = []
        for j in range(len(b[0])):
            element = 0
            for k in range(len(b)):
                element += a[i][k] * b[k][j]
            row.append(element)
        result.append(row)
    return result

# 使用NumPy替换嵌套循环计算两个数组的乘积
def multiply_arrays_with_numpy(a, b):
    return np.dot(a, b)

# 创建两个数组
a = np.array([[1, 2], [3, 4]])
b = np.array([[5, 6], [7, 8]])

# 使用嵌套循环计算两个数组的乘积
result1 = multiply_arrays(a, b)
print("使用嵌套循环计算的结果：")
print(result1)

# 使用NumPy替换嵌套循环计算两个数组的乘积
result2 = multiply_arrays_with_numpy(a, b)
print("使用NumPy计算的结果：")
print(result2)

使用嵌套循环计算的结果：
[[19, 22], [43, 50]]
使用NumPy计算的结果：
[[19 22]
 [43 50]]