空のndarrayを用意したい

前提

pytorchで機械学習を行うためのデータローダを作成しています。

実現したいこと

空のndarrayに.npyファイルから読み込んだ値を一つずつ格納していく。
その時、配列の宣言の際に、data = np.zeros(256)を行うことによって0詰めのいらない一行が入ってしまう。

発生している問題

device : cuda:0 list size : 5584 (5585, 256) data[0]=[0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. … 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0.] data[1]=[-0.59893709 0.80101025 0.86713511 0.27356422 0.03806147 0.55685222 -1.11983168 -0.46897823 -0.27684769 0.1312255 0.30654386 -0.07118449 … 0.57396317 -0.56885028 -0.51682287 -1.14913428 -0.02551528 0.59562939 0.98224401 -0.52175301 -0.33242595 0.01973815 -0.45488593 -0.34487733 0.87223083 1.15911603 0.65931791 1.32030606]

該当のソースコード

python

# PyTorchimport torch import torch.nn as nn import torch.nn.functional as F #import glob import numpy as np batch_size = 128one_z_size = 256 dev = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") print("device :", dev) #データのファイル名取得path = '../../../data/z_out/acoustic_text_feature/'file_path_lists = glob.glob("{}/**".format(path), recursive=True)file_path_lists.pop(0)print("list size :",len(file_path_lists)) data = np.zeros(256)data_name = np.empty(0)for i in range(len(file_path_lists)): data_name = np.append(data_name, file_path_lists[i]) #print(np.load(file_path_lists[i])) data = np.vstack((data, np.load(file_path_lists[i]))) print(data.shape)print(data[0])

試したこと

data = np.zeros(256)のところで一応配列の定義のような作業を行っています。0詰めのデータなので、いらないのですが、これを消すと、dataが定義されていないというエラーがでます。空の配列を宣言したいのですが、そのような記事が見つけることができませんでした。もっとスマートなやり方があったら教えていただきたいです。

コメントを投稿

0 コメント