给定 2*3 A= [[10,20],[74,25],[340,20]] 的矩阵,得到输出 = [[10,30,-10,20],[74,99, 49,25],[340,360,320,20]] 其中第 1 和第 2 列值将是第 1 和第 4 列值,输出的第 2、第 3 列将是第 1 和第 2 列的加法和减法。 python 中的解决方案是什么?
我尝试过转置
您可以使用以下命令执行您的特定操作,但我不知道您想要多通用。
正如其他人评论的那样,“转置”是对您在这里所做的事情的非常不恰当的描述。 (你的原始矩阵也是 3x2,而不是 2x3)。
import numpy as np
A = np.array( [ [ 10, 20 ], [ 74, 25 ], [ 340, 20 ] ] )
B = np.column_stack( ( A[:,0], ( A[:,0] + A[:,1] ), ( A[:,0] - A[:,1] ), A[:,1] ) )
print( A, '\n\n', B )
输出:
[[ 10 20]
[ 74 25]
[340 20]]
[[ 10 30 -10 20]
[ 74 99 49 25]
[340 360 320 20]]