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

从左到右用非零值填充一维NumPy数组的零值?

要从左到右用非零值填充一维NumPy数组的零值,可以使用NumPy中的函数np.where()来实现。np.where()函数可以根据条件选择性地替换数组中的元素。

以下是一个示例代码:

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

def fill_zeros(arr):
    non_zero_values = arr[arr != 0]  # 获取非零值
    zero_indices = np.where(arr == 0)[0]  # 获取零值的索引
    num_zeros = len(zero_indices)  # 零值的数量

    # 如果非零值的数量小于等于零值的数量,则将非零值按顺序填充到零值位置
    if len(non_zero_values) <= num_zeros:
        arr[zero_indices[:len(non_zero_values)]] = non_zero_values
    else:
        # 如果非零值的数量大于零值的数量,则将非零值按循环方式填充到零值位置
        arr[zero_indices] = np.tile(non_zero_values, num_zeros // len(non_zero_values) + 1)[:num_zeros]

    return arr

# 示例用法
arr = np.array([0, 2, 0, 4, 0, 6, 0])
filled_arr = fill_zeros(arr)
print(filled_arr)

输出结果为:

代码语言:txt
复制
[2 2 4 4 6 6 0]

在这个示例中,我们首先使用arr[arr != 0]获取了数组中的非零值,然后使用np.where(arr == 0)[0]获取了零值的索引。接下来,我们判断了非零值的数量和零值的数量的关系,如果非零值的数量小于等于零值的数量,则直接将非零值按顺序填充到零值位置;如果非零值的数量大于零值的数量,则使用np.tile()函数将非零值按循环方式填充到零值位置。最后返回填充后的数组。

这个方法可以适用于任意一维NumPy数组的填充操作。

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

相关·内容

没有搜到相关的结果

领券