Преобразование списка в массив и извлечение определенных координат из преобразованного массива - PullRequest
Новая
       12

Преобразование списка в массив и извлечение определенных координат из преобразованного массива

0 голосов
/ 09 мая 2019

У меня есть вывод в виде списка, в котором мне нужно выполнить некоторые операции.Для этого я преобразовал в массив, используя функцию np.asarray.

Мой список 

[[ 0.00000000e+00,  0.00000000e+00,  0.00000000e+00,
        -5.41511240e+01, -2.30953821e+01, -1.74871288e+01,
        -4.91000564e+01,  2.44382720e+02,  1.71213983e+02,
        -1.54755310e+02,  5.97881714e+02,  3.52366218e+02,
        -1.15560633e+01,  7.42149725e+02,  1.66477287e+02,
        -1.18447102e+01,  7.36763064e+02,  1.65182437e+02,
         5.41502771e+01,  2.30950820e+01,  1.74870470e+01,
        -1.71258190e+01,  3.26592894e+02,  2.03220778e+02,
        -1.41130065e+02,  7.04033786e+02,  3.84201614e+02,
        -1.12642400e+01,  7.48636864e+02,  1.66665977e+02,
        -1.14090840e+01,  7.36435064e+02,  1.63713810e+02,
         1.21660845e-03, -8.60110629e-02, -1.93000576e-02,
         1.57141460e+01, -2.34399996e+02, -2.86722926e+01,
         5.28252697e+01, -4.40469167e+02, -1.11653705e+02,
         1.03085631e+02, -5.01280352e+02, -1.93111585e+02,
         7.16011844e+01, -5.88214725e+02, -2.18615940e+02,
         5.67537804e+00, -4.35088906e+02, -9.76974016e+01,
         7.71909591e+01, -3.88738749e+02, -6.29099586e+01,
        -2.99970496e+01, -2.25985794e+02,  6.41590789e+01,
         1.03847001e+02, -7.32419021e+01,  1.04802558e+02,
         1.26585822e+00, -1.20170579e+02, -2.82526049e+01,
         1.57900698e+00, -1.51780249e+02, -3.52080548e+01,
         8.84543993e-01, -1.07795356e+02, -2.56307189e+01,
         8.84543993e-01, -1.07795356e+02, -2.56307189e+01,
         5.67537804e+00, -4.35088906e+02, -9.76974016e+01,
         8.01141013e+00, -4.16078607e+02, -1.25355227e+02,
         1.17740492e+00, -2.55151916e+02, -7.20503620e+01,
        -1.73992688e+01, -2.44854505e+02, -9.25408725e+01,
         8.70569014e-01, -1.68664569e+02, -3.73902498e+01,
         1.39982512e+00, -2.00884252e+02, -4.47207875e+01,
         5.24591115e-01, -1.65867774e+02, -3.68342864e+01,
         5.24591115e-01, -1.65867774e+02, -3.68342864e+01]])]

Я планировал извлечь первый столбец и создать vector x извлечь второй столбец vector y и третий столбец vector z.Однако когда я преобразовал его в массив, это привело к массиву 1x1x96 со следующей структурой 

[[[ 0.00000000e+00  0.00000000e+00  0.00000000e+00 -5.41511240e+01
   -2.30953821e+01 -1.74871288e+01 -4.91000564e+01  2.44382720e+02
    1.71213983e+02 -1.54755310e+02  5.97881714e+02  3.52366218e+02
   -1.15560633e+01  7.42149725e+02  1.66477287e+02 -1.18447102e+01
    7.36763064e+02  1.65182437e+02  5.41502771e+01  2.30950820e+01
    1.74870470e+01 -1.71258190e+01  3.26592894e+02  2.03220778e+02
   -1.41130065e+02  7.04033786e+02  3.84201614e+02 -1.12642400e+01
    7.48636864e+02  1.66665977e+02 -1.14090840e+01  7.36435064e+02
    1.63713810e+02  1.21660845e-03 -8.60110629e-02 -1.93000576e-02
    1.57141460e+01 -2.34399996e+02 -2.86722926e+01  5.28252697e+01
   -4.40469167e+02 -1.11653705e+02  1.03085631e+02 -5.01280352e+02
   -1.93111585e+02  7.16011844e+01 -5.88214725e+02 -2.18615940e+02
    5.67537804e+00 -4.35088906e+02 -9.76974016e+01  7.71909591e+01
   -3.88738749e+02 -6.29099586e+01 -2.99970496e+01 -2.25985794e+02
    6.41590789e+01  1.03847001e+02 -7.32419021e+01  1.04802558e+02
    1.26585822e+00 -1.20170579e+02 -2.82526049e+01  1.57900698e+00
   -1.51780249e+02 -3.52080548e+01  8.84543993e-01 -1.07795356e+02
   -2.56307189e+01  8.84543993e-01 -1.07795356e+02 -2.56307189e+01
    5.67537804e+00 -4.35088906e+02 -9.76974016e+01  8.01141013e+00
   -4.16078607e+02 -1.25355227e+02  1.17740492e+00 -2.55151916e+02
   -7.20503620e+01 -1.73992688e+01 -2.44854505e+02 -9.25408725e+01
    8.70569014e-01 -1.68664569e+02 -3.73902498e+01  1.39982512e+00
   -2.00884252e+02 -4.47207875e+01  5.24591115e-01 -1.65867774e+02
   -3.68342864e+01  5.24591115e-01 -1.65867774e+02 -3.68342864e+01]]]

Это приводит к массиву [x0,y0,z0,x1,y1,z1,x3,y3,z3...... x31,y31,z31], и я хочу извлечь следующее: 

x = [x0, x2, .......... x31]
y= [y0, y1, ...........y31]
z= [z0,z1,..............z31]

Мне было интересно, возможно ли это сделать в list или изменить array

Ответы [ 2 ]

0 голосов
/ 09 мая 2019

IICU, учитывая массив 

[x0,y0,z0,x1,y1,z1,x2,y2,z2, .... , x31,y31,z31]

Вы можете просто использовать индексирование NumPy 

x = arr[np.arange(0,len(arr),3)]
y = arr[np.arange(1,len(arr),3)]
z = arr[np.arange(2,len(arr),3)]
0 голосов
/ 09 мая 2019

Редактировать:

Как предлагается в субсэмплирование каждой n-й записи в массиве NumPy , Numpy.ndarray предлагает простой и элегантный вариант: 

import numpy as np
arr = np.array(my_list)
x = arr[::3]
y = arr[1::3]
z = arr[2::3]

Оригинальный ответ:

Я не уверен насчет структуры вашего списка ввода, так как вы предоставили здесь неверные данные.

Несмотря надополнительные скобки в исходных данных, это можно сделать с помощью довольно простого трюка, и вам не нужно беспокоиться о Numpy:

Предполагая, что вы пытаетесь каким-то образом разбить список на три векторауказанным способом присвойте свой список a, затем вы можете получить нужные списки по: 

x = [a[3*i] for i in range(int(len(a)/3))]
y = [a[3*i+1] for i in range(int(len(a)/3))]
z = [a[3*i+2] for i in range(int(len(a)/3))]
...