python - 如何修复以下代码？

Question

def CostFunction(A):
    return sum(A)

A = [[1,1,1],[2,2,2]]
print CostFunction(A[0:])

答案应该是3，但应该有问题A。

Answer 1

A[0:]是一个切片，而不是一个元素：

>>> A = [[1,1,1],[2,2,2]]
>>> A[0:]
[[1, 1, 1], [2, 2, 2]]

更一般地，返回从索引到列表末尾的A[n:]元素。有关更多详细信息，请参阅Python 的切片表示法。An

我相信你想要A[0]：

>>> A[0]
[1, 1, 1]