np.where生成的数组
时间: 2023-10-29 07:54:11 浏览: 85
python由已知数组快速生成新数组的方法
通过使用np.where函数,可以生成一个新的数组。np.where函数的使用方式是满足条件时将数组元素修改为x,否则修改为y。生成的新数组将保留原数组的形状和维度,并且不会修改原数组的数值。
举个例子,如果我们有一个一维数组arr1,其中包含了一些正数和负数,我们可以使用np.where(arr1>0, 1, -1)来生成一个新的数组arr2。在arr2中,原数组arr1中大于0的元素会被替换成1,小于等于0的元素会被替换成-1。
另外,np.where函数同样适用于二维数组。如果我们有一个二维数组arr1,我们同样可以使用np.where(arr1>0, 1, -1)来生成一个新的数组arr2。在arr2中,原数组arr1中大于0的元素会被替换成1,小于等于0的元素会被替换成-1。
总之,通过使用np.where函数,我们可以根据条件来生成新的数组,其中满足条件的元素会被修改,而不满足条件的元素会保持不变。<span class="em">1</span><span class="em">2</span><span class="em">3</span>
阅读全文