Question

我将1和0的Pandas数据框转换为数组： [0 0 0 1 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 1]

我正在使用以下函数来修改数组：

def regressor(x, param):
    new_array = x
    for i in range(len(new_array)):

        length = len(new_array)
        current = new_array[(length-i)-1]
        previous = new_array[(length-i)-2]

        if current != 0:
            if previous == 0:
                new_array[(length-i)-2] = current*param

    return new_array

但是，我的数组仍然保持不变。 new_array [（length-i）-2]似乎实际上并未修改该数组的元素。

有人可以告诉我我想念什么吗？谢谢

更新：我的问题已解决，方法是将DataFrame转换为pandas.Series，然后将其转换为函数中的列表。

Answer 1

如果我运行您的代码，我会得到这个（对于param = 2）：

x = [0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1]

def regressor(x, param):
    new_array = x
    for i in range(len(new_array)):

        length = len(new_array)
        current = new_array[(length-i)-1]
        previous = new_array[(length-i)-2]

        if current != 0:
            if previous == 0:
                new_array[(length-i)-2] = current * param

    return new_array

new_array = regressor(x, 2)
print(new_array)
# Output:
# [8, 4, 2, 1, 2097152, 1048576, 524288, 262144, 131072, 65536, 32768, 16384, 8192, 4096, 2048, 1024, 512, 256, 128, 64, 32, 16, 8, 4, 2, 1]

为什么您没有看到new_array中的更改？您是否真的调用了回归函数来更新数组？另外，在函数开始处将x复制到new_array是多余的。

只需重新发布函数并进行一些编辑即可获得更好的可读性：

def regressor(arr, param):
    for i in range(len(arr)):

        length = len(arr)
        current = arr[(length-i)-1]
        previous = arr[(length-i)-2]

        if (current != 0) and (previous == 0):
                arr[(length-i)-2] = current * param

    return arr

如何在for循环中修改Numpy数组

1 个答案: