问题描述
import numpy as np
k1=np.array([[1,-1,1,0],[0,10,25,90],[20,80]])
i=0
while(i<=2):
if(k1[i,i]!=1 and k1[i,i]!=0):
k1[i,:]=k1[i,:]/3.0
print(k1[i,:])
i=i+1
解决方法
您需要提及 dtype
import numpy as np
k1=np.array([[1,-1,1,0],[0,10,25,90],[20,80]],dtype=np.float32)
i=0
while(i<=2):
if(k1[i,i]!=1 and k1[i,i]!=0):
k1[i,:]=k1[i,:]/3.0
print(k1[i,:])
i=i+1
[ 0. 3.3333333 8.333333 30. ]