如何修复下面的代码？

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:]会将A的元素从索引n返回到列表的末尾。有关详细信息，请参阅Python's slice notation。

我相信你想要A[0]：

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