import numpy as np
a=np.arange(1,13,1)
a=np.reshape(a,(3,4))
输出
[[ 1 2 3 4]
[ 5 6 7 8]
[ 9 10 11 12]]
首先创建一个3行4列的数组
a[...,-1][a[...,-1]>1]=0
print(a)
'''输出'''
[[ 1 2 3 0]
[ 5 6 7 0]
[ 9 10 11 0]]
使最后一列大于1的数值为0
例题:创建一个4行4列的数组([[1 2 3 4] 使数组的第二行到第四行,第二列到第四列的奇数为0 [5 6 7 8]
[9,10,11,12]
[13,14,15,16]])
解
import numpy as np
a=np.arange(1,17,1)
a=np.reshape(a,(4,4))
print(a)#修改前
a[1:,1:][a[1:,1:]%2!=0]=0
print(a)#修改后
输出