Question

我有以下问题，如下所述。如何获得锻炼所需的输出？

您的目标是完成一个函数output = matrix_manip(A, B)，其中A和B是输入矩阵（由np.array表示）。 matrix_manip函数应返回一个python字典，其中包含以下描述的操作结果。要使用一些数据，请使用以下矩阵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']。我怎样才能达到所需的结果？

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]])

Answer 2

好像您在问如何使用字典，不是吗？

您只需要创建dict，输入所需的值，然后将其返回即可。

def matrix_manip(A, B):
    results = {}
    results['A_transpose'] = np.transpose(A)
    # ... More code here ...
    return results

output = matrix_manip(A, B)

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

2 个答案: