Как сохранить подмассив в файле npy? - PullRequest
Новая
       8

Как сохранить подмассив в файле npy?

0 голосов
/ 10 февраля 2019

Мои треки данных имеют следующую форму:

(13044,). Типы данных: 

tracks.dtype.names

('frame_num','mean_x','mean_y','var_x','var_y', 'length', 'scale', 'x_pos','y_pos', 't_pos', 'coords', 'trajectory', 'hog', 'hof', 'mbh_x','mbh_y')

dtype([('frame_num', '<i4'), ('mean_x', '<f4'), ('mean_y', '<f4'), ('var_x', '<f4'), ('var_y', '<f4'), ('length', '<f4'), ('scale', '<f4'), ('x_pos', '<f4'), ('y_pos', '<f4'), ('t_pos', '<f4'), ('coords', '<f4', (16, 2)), ('trajectory', '<f4', (15, 2)), ('hog', '<f4', (96,)), ('hof', '<f4', (108,)), ('mbh_x', '<f4', (96,)), ('mbh_y', '<f4', (96,))])

Мне нужно сохранить (16,2) координаты данных 13044 в новый файл npy.,Я попробовал следующее: 

x=tracks['coords']
for i in range(0,len(tracks)):
    y=tracks['coords'][i]
    print(y)
    np.save('test.npy',y)

Вывод y: там данные 13044, я поставил только несколько 

[[182.        92.      ]
 [182.54565   92.09981 ]
 [183.10211   91.61575 ]
 [183.64021   92.13559 ]
 [184.27351   92.15997 ]
 [185.0328    92.20285 ]
 [185.6495    92.19383 ]
 [185.88063   92.225876]
 [186.3553    92.30736 ]
 [187.29843   92.38876 ]
 [187.89871   92.38898 ]
 [188.25539   92.46452 ]
 [188.98816   92.39856 ]
 [189.5047    92.37273 ]
 [189.76077   92.67736 ]
 [190.50615   92.31434 ]]
[[187.        92.      ]
 [187.56187   92.08742 ]
 [188.12775   91.60125 ]
 [188.64186   91.94049 ]
 [189.10121   91.90893 ]
 [189.9543    92.00123 ]
 [190.43088   92.01...

..... goes on until end 
[[265.87213 209.30359]
 [266.8972  208.9946 ]
 [267.89746 208.38165]
 [268.8108  207.88152]
 [269.64877 207.46448]
 [270.36688 207.13185]
 [271.16782 206.77945]
 [271.74063 206.21416]
 [272.45694 205.88182]
 [273.10373 205.73294]
 [273.6556  205.66495]
 [274.32462 205.54205]
 [275.11664 205.4512 ]
 [276.0263  205.37993]
 [276.99155 205.18765]
 [277.99423 205.0822 ]]

Пока я загружаю test.npy, он не сохраняет все y .Itпросто сохраните последний массив координат: 

data='test.npy'
data1=np.load(data)
data1

Вывод данных1: 

array([[265.87213, 209.30359],
       [266.8972 , 208.9946 ],
       [267.89746, 208.38165],
       [268.8108 , 207.88152],
       [269.64877, 207.46448],
       [270.36688, 207.13185],
       [271.16782, 206.77945],
       [271.74063, 206.21416],
       [272.45694, 205.88182],
       [273.10373, 205.73294],
       [273.6556 , 205.66495],
       [274.32462, 205.54205],
       [275.11664, 205.4512 ],
       [276.0263 , 205.37993],
       [276.99155, 205.18765],
       [277.99423, 205.0822 ]], dtype=float32)

Как сохранить все (16,2) координаты данных 13044?

1 Ответ

0 голосов
/ 10 февраля 2019

Вы можете сохранить массив всех ваших (16,2) лет: 

ys = []
for i in range(len(tracks)):
    y=tracks['coords'][i]
    ys.append(y)
    print(y)
np.save('test.npy',np.array(ys))

Также обратите внимание, что вы двигаетесь по tracks с i, но читаете tracks['coord'][i] каждыйвремя (может быть разной длины).Кроме того, если не все y имеют одинаковую длину, то было бы проблемой создать из них массив numpy, но вы все равно можете сохранить список через np.save, а при загрузке использовать np.load('test.npy').item() (опять же, есливы не сохраняете их как np.array).

...