Python/numpy 设置数组某一整行列的值为0/1 修改数组行列设置为特定值

使用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]]