使用Python函数从数组中删除零

Question

这是交易，我有一个多个元素的数组，其中大约一半是零。我想通过使用函数而不是传统的x=x[x!=0]来删除这些零。

我试过了：

def funct(x,y):
    x=x[x!=0]
    y=y[y!=0]

但是我得到的输出是我执行函数之前的变量。一个有多个零的数组。

我是python的新手很抱歉，如果这个问题听起来很荒谬。

非常感谢！

Answer 1

x[x!=0]返回一个新数组，并将该新数组分配给局部变量x。

你可以这样做：

def funct(x, y):
    x = x[x!=0]
    y = y[y!=0]
    # do something here
    return x,y
a, b = funct(a, b)  #assign the returned value back to the global variables