我有一个可以是任何深度或长度的列表。我的意思是我可以有一个这样的列表:
lst = [1,2,3]
要么:
lst = [[2,233],[[[4,5],[66.33]],[[24,88.65,103,2200.0],[-44.2,-8,5]]], [[[[[[[[5]]]]]]]]]
而我想做的是随机修改列表中任何一个数值中的一个。我知道我可以通过将列表转换为字符串来做一些狡猾的事情,但是如果有一种标准的方法可以做到这一点,那么与此相关的答案将不胜感激!
编辑:
对于那些不知道的人,你不能简单地随机选择任何这些值并对其进行修改(例如,添加1),因为列表可以嵌套。这是我想要获得的输入和输出的示例:
lst = [[2,233],[[[4,5],[66.33]],[[24,88.65,103,2200.0],[-44.2,-8,5]]], [[[[[[[[5]]]]]]]]]
lst = modify(lst,4) # Where 4 is the amount to add to a random number in the list
>lst: [[2,233],[[[4,9],[66.33]],[[24,88.65,103,2200.0],[-44.2,-8,5]]], [[[[[[[[5]]]]]]]]]
# the fourth number to appear left-to-right in the list 5 has had 4 added to it, ultimately resulting in 9
# this number was randomly selected
再次运行相同的代码,lst
现已更新:
lst = modify(lst,-2)
>lst: [[2,233],[[[4,9],[66.33]],[[24,86.65,103,2200.0],[-44.2,-8,5]]], [[[[[[[[5]]]]]]]]]
# The seventh number 88.65 has had 2 subtracted from it, to ultimately equal 86.65
第一个问题是按顺序迭代列表,无论嵌套有多深。这是一个只返回的生成器(灵感来自this answer):
import functools
import operator
def iter_nested_list(input_list):
# build index of first level elements
index_list_to_check = [(i, ) for i in range(len(input_list))]
while len(index_list_to_check) > 0:
current_index = index_list_to_check.pop(0)
# get the element
elem = functools.reduce(operator.getitem, current_index, input_list)
if isinstance(elem, list):
for i in range(len(elem)):
# this is a list, so we need to check one level deeper
index_list_to_check.append(current_index + (i, ))
else:
# this is not a list, so we yield the index
yield current_index
这可以这样使用:
>>> list_1 = [[2,233],[[[4,5],[66.33]],[[24,88.65,103,2200.0],[-44.2,-8,5]]],[[[[[[[[5]]]]]]]]]
>>> iter_nested_list(list_1)
<generator object iter_nested_list at 0x7fdbbc29d990>
>>> list(iter_nested_list(list_1))
[(0, 0), (0, 1), (1, 0, 0, 0), (1, 0, 0, 1), (1, 0, 1, 0), (1, 1, 0, 0), (1, 1, 0, 1), (1, 1, 0, 2), (1, 1, 0, 3), (1, 1, 1, 0), (1, 1, 1, 1), (1, 1, 1, 2), (2, 0, 0, 0, 0, 0, 0, 0, 0)]
要从列表中获取单个元素,我们可以使用生成的索引:
>>> index_list = list(iter_nested_list(list_1))
>>> index = index_list[1]
>>> index
(0, 1)
>>> functools.reduce(operator.getitem, index, input_list)
233
现在,修改一个元素:
def modify(input_list, value_to_add):
index_list = list(iter_nested_list(list_1))
index = random.choice(index_list)
index_base = index[:-1] # list of all elements from 'index' excluding the last one
index_elem = index[-1] # single element, the last of the list 'index'
# get list that holds the value we randomly selected
sub_list = functools.reduce(operator.getitem, index_base, input_list)
# modify value
sub_list[index_elem] += value_to_add
在这里,它正在行动:
>>> list_1 = [[2,233],[[[4,5],[66.33]],[[24,88.65,103,2200.0],[-44.2,-8,5]]],[[[[[[[[5]]]]]]]]]
>>> modify(list_1, 5)
>>> list_1
[[2, 233], [[[4, 5], [66.33]], [[24, 88.65, 103, 2200.0], [-44.2, -8, 10]]], [[[[[[[[5]]]]]]]]]
>>> modify(list_1, 5)
>>> list_1
[[2, 233], [[[4, 5], [66.33]], [[24, 88.65, 103, 2205.0], [-44.2, -8, 10]]], [[[[[[[[5]]]]]]]]]
>>> modify(list_1, 5)
>>> list_1
[[2, 233], [[[4, 5], [66.33]], [[24, 88.65, 103, 2205.0], [-39.2, -8, 10]]], [[[[[[[[5]]]]]]]]]