Python迭代数组

Question

我一直在尝试遍历一个数组。

下面是代码。

x = ['lemon', 'tea', 'water', ]

def randomShuffle (arr,n):
    from random import choices
    newList=[]


    for item in arr:
        r=choices(arr, k=n)

        if r.count(item) <= 2:
            newList.append(item)

        return (newList)

我想知道写它的逻辑。

谢谢大家

Answer 1

使用 while 循环：如果每个项目要出现两次，那么结果数组的长度应该是输入数组长度的两倍。
当然，请检查不要在结果中添加相同的项目超过两次；）

Choices 返回一个大小为 1 的列表，所以我使用 [0] 来获取元素

xx = ["a", "b", "c"]

def my_function(x):
    res = []
    while len(res) < len(x) * 2:
        c = choices(x)[0]
        if res.count(c) < 2:
            res.append(c)
    return res

my_function(xx)
> ['c', 'c', 'a', 'b', 'a', 'b']
my_function(xx)
> ['a', 'b', 'b', 'a', 'c', 'c']