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

问题描述 投票:0回答:2

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

您的目标是完成函数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']。如何获得所需的结果?

python python-3.x
2个回答
0
投票
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]])

0
投票

似乎您在问如何使用字典,不?

您只需要创建字典,输入所需的值,然后将其返回。

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

output = matrix_manip(A, B)
© www.soinside.com 2019 - 2024. All rights reserved.