python - 声明一个带有数组参数的python函数并将一个数组参数传递给函数调用？

Question

我是 python 的新手，并试图将数组作为参数传递给 python 函数，该函数将列表/数组声明为参数。

我确定我说错了，

开始：

def dosomething(listparam):
         #do something here
dosomething(listargument)

显然这不起作用，我做错了什么？

谢谢

Answer 1

你所拥有的一切都在正确的轨道上。

def dosomething( thelist ):
    for element in thelist:
        print element

dosomething( ['1','2','3'] )
alist = ['red','green','blue']
dosomething( alist )  

产生输出：

1
2
3
red
green
blue

鉴于您的上述评论，有几点需要注意：与 C 系列语言不同，您通常不需要在迭代列表时费心跟踪索引，除非索引本身很重要。但是，如果您确实需要索引，则可以使用enumerate(list)来获得index,element对，而不是x in range(len(thelist))跳舞。

Answer 2

也许你想解包数组的元素，我不知道我是否明白了，但下面是一个例子：

def my_func(*args):
    for a in args:
        print(a)

my_func(*[1,2,3,4])
my_list = ['a','b','c']
my_func(*my_list)

Answer 3

我想我不清楚 OP 的真正要求是什么……你想传递整个数组/列表并在函数内部对其进行操作吗？或者您是否希望对数组/列表中的每个值/项执行相同的操作。如果后者是您希望的，我已经找到了一种行之有效的方法。

我更熟悉 Fortran 和 C 等编程语言，您可以在其中定义元素函数，这些函数对数组中的每个元素进行操作。我终于找到了与此等效的python，并认为我会在这里重新发布解决方案。关键是对函数进行“矢量化”。这是一个例子：

def myfunc(a,b):
    if (a>b): return a
    else: return b
vecfunc = np.vectorize(myfunc)
result=vecfunc([[1,2,3],[5,6,9]],[7,4,5])
print(result)

输出：

[[7 4 5]
 [7 6 9]]