Question

我有一个清单清单。如何创建将10实例随机替换为另一个数字的函数？

lst = [ [10, 100, 10 ], [10, 1000, 10], [100, 100000, 100], [10, 10, 0] ]

这是我的代码。我只能设法弄平列表而不是嵌套列表。如何获取嵌套列表，还有更有效的方法吗？

import random

def random_replace10(num, lst): # replaces 10 with num
    indexes = []
    flattened = [num for row in lst for num in row]
    for ind, ele in enumerate(flattened):
        if ele == 10:
            indexes.append(ind)

    random_index = random.sample(indexes, 1)[0]
    flattened[random_index] = num
    return flattened           #Returns a flattened list instead of a nested list.


print(random_replace10(2, lst))

>>> [ [10, 100, 2 ], [10, 1000, 10], [100, 100000, 100], [10, 10, 0] ]

print(random_replace10(91, lst))

>>> [ [10, 100, 2 ], [10, 1000, 10], [100, 100000, 100], [91, 10, 0] ]

Answer 1

您首先需要找到多少个10值，然后选择用random.randrange()选项替换哪个值。我使用itertools.count()来保持运行中的计数器与所选实例匹配：

from itertools import count
import random


def random_replace10(num, lst):
    selected = random.randrange(sum(sum(1 for i in sub if i == 10) for sub in lst))
    counter = count()
    return [
        [num if i == 10 and next(counter) == selected else i for i in sub] 
        for sub in lst
    ]

此操作将10的一个num值恰好替换为None，并且不再重复，并且创建一个 new 列表。旧列表保持不变。

您也可以就地进行替换，但是在这种情况下，我将返回list（就像list.append()上的其他变异方法一样，例如10等））。在这种情况下，请首先记录random.choice()值的位置，然后使用import random def random_replace10_inplace(num, lst): locations = [(i, j) for i, sub in enumerate(lst) for j, v in enumerate(sub) if v == 10] outer, inner = random.choice(locations) lst[outer][inner] = num选择其中一个位置：

>>> lst = [[10, 100, 10], [10, 1000, 10], [100, 100000, 100], [10, 10, 0]]
>>> random_replace10(2, lst)
[[10, 100, 10], [10, 1000, 2], [100, 100000, 100], [10, 10, 0]]
>>> random_replace10(2, lst)
[[10, 100, 10], [10, 1000, 10], [100, 100000, 100], [10, 2, 0]]
>>> random_replace10(91, lst)
[[10, 100, 91], [10, 1000, 10], [100, 100000, 100], [10, 10, 0]]
>>> random_replace10(91, lst)
[[10, 100, 10], [10, 1000, 91], [100, 100000, 100], [10, 10, 0]]

第一个版本的演示

>>> lst = [[10, 100, 10], [10, 1000, 10], [100, 100000, 100], [10, 10, 0]]
>>> random_replace10_inplace(2, lst)
>>> lst
[[10, 100, 10], [10, 1000, 2], [100, 100000, 100], [10, 10, 0]]
>>> random_replace10_inplace(2, lst)
>>> lst
[[10, 100, 10], [2, 1000, 2], [100, 100000, 100], [10, 10, 0]]
>>> random_replace10_inplace(91, lst)
>>> lst
[[10, 100, 10], [2, 1000, 2], [100, 100000, 100], [91, 10, 0]]
>>> random_replace10_inplace(91, lst)
>>> lst
[[91, 100, 10], [2, 1000, 2], [100, 100000, 100], [91, 10, 0]]

和就地版本：

{{1}}

Answer 2

您可以执行以下操作

import random

lst = [[10, 100, 10], [10, 1000, 10], [100, 100000, 100], [10, 10, 0]]


def random_replace10(num, lst):
    ten_index = []
    for i, item in enumerate(lst):
        for j, inner_item in enumerate(item):
            if inner_item == 10:
                ten_index.append((i, j))

    i, j = random.choice(ten_index)
    lst[i][j] = num
    print(lst)

如何随机替换嵌套列表中的特定元素？

2 个答案: