使用numpy中fill函数将数组某一行所有值修改为特定值
array_col.fill(value):将数组的第col列的值全部修改为value
array_col: 你要修改的列
value:你要修改为的值
1. 预备知识 – 数组列/行的获取,以二维数组为例
获取第i行: array[i]
获取第j列: array[:,j]
注:行列皆为从第0行开始
import numpy as np
array = np.arange(0,20).reshape(4,5)
print("数组值为",array)
print("第2行:",array[2])
print("第3列",array[:,3])
输出结果:
数组值为 [[ 0 1 2 3 4]
[ 5 6 7 8 9]
[10 11 12 13 14]
[15 16 17 18 19]]
第2行: [10 11 12 13 14]
第3列 [ 3 8 13 18]
2 修改行列值
import numpy as np
array = np.arange(0,20).reshape(4,5)
# 修改第2行的值全为0
array[2].fill(0)
print(array)
print('-----------------------')
# 修改第3列的值全为5
array[:,3].fill(5)
print(array)
输出结果:
[[ 0 1 2 3 4]
[ 5 6 7 8 9]
[ 0 0 0 0 0]
[15 16 17 18 19]]
-----------------------
[[ 0 1 2 5 4]
[ 5 6 7 5 9]
[ 0 0 0 5 0]
[15 16 17 5 19]]