作者:
TuCH (謬客)
2018-10-05 16:26:55※ 引述《sssh (叫我松高魂 ~~)》之銘言:
: 想請問一下,這幾天碰到一個問題,
: 有兩個相同大小(3*30)的矩陣B,C
: 我想要弄出一個新的矩陣D
: 新矩陣裡面的元素是給定兩矩陣中相對應位置元素取大值
: 於是我寫了以下
: for i in range(2):
: for j in range(29):
: def D(i,j):
: if (B[i][j]<C[i][j]):
: return C[i][j]
: else:
: return B[i][j]
: 但是問題來了........怎麼把D(i,j)變成D[i][j]啊@@
: 怎麼寫都出現語法錯誤XDDDDDD
: 小弟在此先感謝各位大德XD
你的寫法
import numpy as np
B = np.random.random([3,30])
C = np.random.random([3,30])
D = np.zeros([3,30])
for i in range(3):
for j in range(30):
if (B[i][j]<C[i][j]):
D[i][j]=C[i][j]
else:
D[i][j]=B[i][j]
另外一種
import numpy as np
B = np.random.random([3,30])
C = np.random.random([3,30])
B[B < C]=0
C[B != 0] =0
D = B+C