可以使用numpy的条件索引来对数组中的值进行条件化。以下是一个示例代码:
import numpy as np
# 创建一个随机数组
arr = np.random.randint(0, 10, size=(5, 5))
print("原始数组:")
print(arr)
# 根据条件对数组中的值进行条件化
condition = arr > 5
arr[condition] = 10
arr[~condition] = 0
print("条件化后的数组:")
print(arr)
输出结果:
原始数组:
[[8 2 6 2 3]
[1 3 1 2 7]
[3 8 1 9 6]
[6 8 3 4 9]
[2 8 2 6 3]]
条件化后的数组:
[[10 0 10 0 0]
[ 0 0 0 0 10]
[ 0 10 0 10 10]
[10 10 0 0 10]
[ 0 10 0 10 0]]
在示例代码中,我们首先创建一个随机数组arr。然后,我们使用条件arr > 5创建一个布尔类型的条件索引。接下来,我们使用条件索引将大于5的元素设置为10,将小于等于5的元素设置为0。最后,我们打印出条件化后的数组。
上一篇:按周创建时间分区的