In [3]:
import numpy as np
a3 = np.random.randint(0, 10, size=(3, 5))
print(a3)
[[6 3 0 0 0] [8 5 7 2 4] [7 5 6 9 0]]
In [4]:
# 3.1使用索引进行替换
# 第2行数据替换为0
a3[1] = 0
print(a3)
[[6 3 0 0 0] [0 0 0 0 0] [7 5 6 9 0]]
In [5]:
# 3.2使用array进行替换
# 第2行数据替换为1,2,3,4,5
a3[1] = np.array([1,2,3,4,5])
print(a3)
[[6 3 0 0 0] [1 2 3 4 5] [7 5 6 9 0]]
In [6]:
# 3.3使用条件进行索引替换
# 元素值小于3的都替换为1
a3[a3 < 3] = 1
print(a3)
[[6 3 1 1 1] [1 1 3 4 5] [7 5 6 9 1]]
In [7]:
# 3.4使用where函数替换: 把符合条件的值替换为x，剩余替换为y
result = np.where(a3 < 5, 0 , 1) # a3中所有小于5的替换为0，剩余替换为1
print(a3)
print(result)
[[6 3 1 1 1] [1 1 3 4 5] [7 5 6 9 1]] [[1 0 0 0 0] [0 0 0 0 1] [1 1 1 1 0]]
