我有以下问题,如下所述。如何获得锻炼所需的输出?
您的目标是完成函数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']
。如何获得所需的结果?
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]])
似乎您在问如何使用字典,不?
您只需要创建字典,输入所需的值,然后将其返回。
def matrix_manip(A, B):
results = {}
results['A_transpose'] = np.transpose(A)
# ... More code here ...
return results
output = matrix_manip(A, B)