Pythonic方法生成可能的数组列表

Question

给定一个数组（例如[3,5,2]），我试图生成一个可能的数组列表，这些数组是在数组中添加1到1个条目得到的：[[4,5,2]，[3 ，6,2]，[3,5,3]]。

我可以通过以下方式完成它，但想知道是否有更多的pythonic方式来获得结果？

test = [3, 5, 2]
result = [t.copy() for _ in range(len(test))]
for index, _ in enumerate(result):
    result[index][index] += 1

Answer 1

以下是如何使用列表理解来完成它：

test = [3, 5, 2]   
print [test[:i] + [v + 1] + test[i+1:] for i,v in enumerate(test)]

产量

[[4, 5, 2], [3, 6, 2], [3, 5, 3]]

Answer 2

这是列表理解的另一个内联解决方案：

test = [3, 5, 2]
result = [[v+1 if i==j else v for j, v in enumerate(test)] for i in range(len(test))]

或者，正如PM 2Ring所注意到的，你可以利用True == 1和False == 0的事实：

result = [[v + (i==j) for j, v in enumerate(test)] for i in range(len(test))]