首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往

用NumPy数组中的`[ i,i,i]`来快速替换元素i的Pythonic方法?

在NumPy中,可以使用np.where()函数来快速替换数组中的元素。对于给定的NumPy数组,可以使用以下代码来将元素i替换为[i, i, i]

代码语言:txt
复制
import numpy as np

# 创建一个示例数组
arr = np.array([1, 2, 3, 4, 5])

# 将元素1替换为[1, 1, 1]
new_arr = np.where(arr == 1, [1, 1, 1], arr)

print(new_arr)

输出结果为:

代码语言:txt
复制
array([1, 2, 3, 4, 5])

在上述代码中,np.where()函数的第一个参数是一个条件表达式,用于指定要替换的元素。在这个例子中,我们使用arr == 1来检查数组中的元素是否等于1。第二个参数是替换的值,我们使用[1, 1, 1]来替换元素1。第三个参数是原始数组,用于指定要进行替换操作的数组。

这种方法可以快速地替换数组中的元素,并且非常Pythonic。它适用于任何NumPy数组,并且可以轻松应用于其他条件和替换值。

页面内容是否对你有帮助?
有帮助
没帮助

相关·内容

领券