在python中的n个交替的1和-1的数组

时间:2017-08-23 06:36:56

标签: python

我将如何创建一个重复n 1&1和-1的数组。 即数组[1,-1,1,-1,1,-1,...] n次

我尝试使用numpy.repeat,但它重复了每个元素,而不是一遍又一遍地重复1和-1。

编辑:这是我尝试过的:

import numpy as np

def repeat(n):
    z=np.arange(1,2)
    x=-np.arange(1,2)
    w=np.concatenate((x,z))
    return np.repeat(w,n)

,其中n = 3返回

 array([-1, -1, -1,  1,  1,  1])

理想情况下,我希望这会返回

 array([-1, 1, -1, 1, -1, 1])

1 个答案:

答案 0 :(得分:1)

alternating_list = [1,-1] * number_of_iterations