Нужна помощь сращивания списка - PullRequest
Новая
       12

Нужна помощь сращивания списка

0 голосов
/ 16 января 2020

У меня есть список, скажем, размера 144. Мне нужно, чтобы каждые 16 элементов хранились в двумерном массиве. В конце я хотел бы иметь 16x9 2-D массив. Как мне вытащить эти элементы в al oop и переставить их? Вот что у меня есть ... 

yuH = np.zeros([N, t])
i = 0
j = 0
for i in range(t):
    for j in range(len(p)):
        print(p[0+j:16+j])
        yuH[:,i] = p[0+j:16+j] # change to N
        j = j + 16

Где N в этом случае равно 16, а t равно 9. p - мой список 144. yuH - матрица, которую я пытаюсь создать. увы это не работает ... Я получаю ValueError "установка элемента массива с последовательностью" в строке yuH [:, i] = p [0 + j: 16 + j] после многих запусков.

Что-то вроде ... 

p = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20]

p = [[1,  [5,  9,  [13, [17, 
      2,   6,  10,  14,  18, 
      3,   7,  11,  15,  19, 
      4],  8]  12], 16], 20]]

1 Ответ

1 голос
/ 16 января 2020

Если вам нужно просто реорганизовать ваш массив p, вы можете просто сделать это: 

import numpy as np

N = 16
t = 9
p = np.arange(t*N) # example array p

yuH = p.reshape([t,N])

print(yuH.shape)

>>> (9, 16)

print(yuH)

>>> 
[[  0   1   2   3   4   5   6   7   8   9  10  11  12  13  14  15]
 [ 16  17  18  19  20  21  22  23  24  25  26  27  28  29  30  31]
 [ 32  33  34  35  36  37  38  39  40  41  42  43  44  45  46  47]
 [ 48  49  50  51  52  53  54  55  56  57  58  59  60  61  62  63]
 [ 64  65  66  67  68  69  70  71  72  73  74  75  76  77  78  79]
 [ 80  81  82  83  84  85  86  87  88  89  90  91  92  93  94  95]
 [ 96  97  98  99 100 101 102 103 104 105 106 107 108 109 110 111]
 [112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127]
 [128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143]]
...