如何按Python顺序组合三个不同大小的列表

时间:2017-03-29 13:56:20

标签: python string list for-loop zip

我有三个列表:signal,returns和pct_list:

     signal= ['signal0', 'signal1', 'signal2']

     returns = ['retSCIENCE_MSFT','retLOVE_MSFT','retMUSIC_MSFT',
                 'retSCIENCE_YHOO', 'retLOVE_YHOO', 'retMUSIC_YHOO']

     pct_list = ['Pct_MSFT', 'Pct_YHOO']

我尝试使用zip功能,它给了我以下结果:

    ret_sig_list = list(zip(returns, signal, pct_list))
    ret_sig_list

输出:

     [('retSCIENCE_MSFT', 'signal0', 'Pct_MSFT'),
      ('retLOVE_MSFT', 'signal1', 'Pct_YHOO')]

我想要得到的是MSFT和YHOO都排列了相应的信号和百分比变化如下:

ret_sig_list = [('retSCIENCE_MSFT', 'signal0', 'Pct_MSFT'),
               ('retLOVE_MSFT', 'signal1', 'Pct_MSFT'),
               ('retMUSIC_MSFT', 'signal2', 'Pct_MSFT'),
               ('retSCIENCE_YHOO', 'signal0', 'Pct_YHOO'),
               ('retLOVE_YHOO', 'signal1', 'Pct_YHOO'),
               ('retMUSIC_YHOO','signal2', 'Pct_YHOO')]

我尝试使用list(zip(returns,signal,pct_list)),但它似乎默认为三个列表中最小的元素数,即pct_list。

如果有人知道如何解决这个问题,我会非常感激! 我想使用带有for循环的ret_sig_list来对多个股票执行计算百分比变化上升或下降以根据相关单词给出回报;科学,爱情,音乐等 我从谷歌趋势的Python程序包中得到了一个名为Pytrends的文章,我推荐它,非常酷!

2 个答案:

答案 0 :(得分:3)

使用itertools.cycle

In [2]: from itertools import cycle

In [3]: list(zip(returns, cycle(signal), cycle(pct_list)))
Out[3]:
[('retSCIENCE_MSFT', 'signal0', 'Pct_MSFT'),
 ('retLOVE_MSFT', 'signal1', 'Pct_YHOO'),
 ('retMUSIC_MSFT', 'signal2', 'Pct_MSFT'),
 ('retSCIENCE_YHOO', 'signal0', 'Pct_YHOO'),
 ('retLOVE_YHOO', 'signal1', 'Pct_MSFT'),
 ('retMUSIC_YHOO', 'signal2', 'Pct_YHOO')]

答案 1 :(得分:1)

所以我设法找到了一个解决方案,它并不理想,但很接近。我说不理想的原因不包括["科学","爱","音乐和#34;]列表中的具体单词,但它是唯一的创建有序邮政列表的方法。

  stock_list = ['MSFT', 'YHOO'] 
  stock_list_v2 = []

     for stock in stock_list:
          for i in range(10):
           s = 'ret' + str(i) + '_' + stock

           stock_list_v2.append(s)    
  stock_list_v2   

输出:

  ['ret0_MSFT',
  'ret1_MSFT',
  'ret2_MSFT',
  'ret3_MSFT',
  'ret4_MSFT',
  'ret5_MSFT',
  'ret6_MSFT',
  'ret7_MSFT',
  'ret8_MSFT',
  'ret9_MSFT',
  'ret0_YHOO',
  'ret1_YHOO',
  'ret2_YHOO',
  'ret3_YHOO',
  'ret4_YHOO',
  'ret5_YHOO',
  'ret6_YHOO',
  'ret7_YHOO',
  'ret8_YHOO',
  'ret9_YHOO']

下一个信号:

  sig = list(range(10))
  sig
  signal = ["signal{}".format(*sig) for sig in enumerate(sig)]
  signal

输出:

  ['signal0',
   'signal1',
   'signal2',
   'signal3',
   'signal4',
   'signal5',
   'signal6',
   'signal7',
   'signal8',
   'signal9']

接下来,pct。

   stock_list = ['MSFT', 'YHOO'] 
   stock_list_v3 = []

   for stock in stock_list:
         for i in range(10):

         p = 'pct' + str(i) + '_' + stock
        stock_list_v3.append(p)
  stock_list_v3

输出:

   ['pct0_MSFT',
    'pct1_MSFT',
    'pct2_MSFT',
    'pct3_MSFT',
    'pct4_MSFT',
    'pct5_MSFT',
    'pct6_MSFT',
    'pct7_MSFT',
    'pct8_MSFT',
    'pct9_MSFT',
    'pct0_YHOO',
    'pct1_YHOO',
    'pct2_YHOO',
    'pct3_YHOO',
    'pct4_YHOO',
    'pct5_YHOO',
    'pct6_YHOO',
    'pct7_YHOO',
    'pct8_YHOO',
    'pct9_YHOO']

现在我们从itertools导入循环。

     from itertools import cycle

     full_list = list(zip(stock_list_v2, cycle(signal), stock_list_v3))

     full_list

输出:

     [('ret0_MSFT', 'signal0', 'pct0_MSFT'),
      ('ret1_MSFT', 'signal1', 'pct1_MSFT'),
      ('ret2_MSFT', 'signal2', 'pct2_MSFT'),
      ('ret3_MSFT', 'signal3', 'pct3_MSFT'),
      ('ret4_MSFT', 'signal4', 'pct4_MSFT'),
      ('ret5_MSFT', 'signal5', 'pct5_MSFT'),
      ('ret6_MSFT', 'signal6', 'pct6_MSFT'),
      ('ret7_MSFT', 'signal7', 'pct7_MSFT'),
      ('ret8_MSFT', 'signal8', 'pct8_MSFT'),
      ('ret9_MSFT', 'signal9', 'pct9_MSFT'),
      ('ret0_YHOO', 'signal0', 'pct0_YHOO'),
      ('ret1_YHOO', 'signal1', 'pct1_YHOO'),
      ('ret2_YHOO', 'signal2', 'pct2_YHOO'),
      ('ret3_YHOO', 'signal3', 'pct3_YHOO'),
      ('ret4_YHOO', 'signal4', 'pct4_YHOO'),
      ('ret5_YHOO', 'signal5', 'pct5_YHOO'),
      ('ret6_YHOO', 'signal6', 'pct6_YHOO'),
      ('ret7_YHOO', 'signal7', 'pct7_YHOO'),
      ('ret8_YHOO', 'signal8', 'pct8_YHOO'),
      ('ret9_YHOO', 'signal9', 'pct9_YHOO')]

希望能帮助任何对如何解决它感到好奇的人:) 再次感谢PrestonM建议算法,Juapana帮助建议使用itertools循环。