python - 如何根据示例获取代码输出？

Question

我有以下问题，如下所述。我怎样才能得到锻炼中需要的输出？

您的目标是完成一个函数output = matrix_manip(A, B)，其中 A 和 B 是输入矩阵（由 np.array 表示）。该matrix_manip函数应返回一个包含下述操作结果的 python dict。要使用一些数据，让我们使用以下矩阵 A：

A = np.array([[16,  2,  3, 13],
              [ 5, 11, 10,  8],
              [ 9,  7,  6, 12],
              [ 4, 14, 15,  1]])

找到矩阵 A 的转置并将其返回output['A_transpose']。示例结果：

>>> output['A_transpose']
array([[16,  5,  9,  4],
       [ 2, 11,  7, 14],
       [ 3, 10,  6, 15],
       [13,  8, 12,  1]])

我对使用 numpy 函数没有任何问题。我尝试使用以下结构来做到这一点：

def A_transpose(A):
    return np.transpose(A)

但这可以正确解决，但是当然不适用于 output['A_transpose']. 我怎样才能达到所需的结果？

标签： pythonpython-3.x

Answer 1

import numpy as np
A = np.array([[16, 2, 3, 13],
[ 5, 11, 10, 8],
[ 9, 7, 6, 12],
[ 4, 14, 15, 1]])


def A_transpose(A):
    return np.transpose(A)

output = dict()
output['A_transpose'] = A_transpose(A)

>>> output['A_transpose']
array([[16,  5,  9,  4],
       [ 2, 11,  7, 14],
       [ 3, 10,  6, 15],
       [13,  8, 12,  1]])