np.vstack填充空数组:
- 建立空数组array
- 利用for语句产生数据块data
- 利用np.expand_dims扩展数据块的维度
- 利用np.vstack([array,data])将data填入array。
for j in range(batches_per_epoch):
train_batch = np.array([]).reshape(0, sequence_length, n_components)#这里定义一个空的np.array
for b in range(batch_size): #这里建立一个for循环语句
index = j*batch_size+b
single_engine = training_set[index]
single_engine = np.expand_dims(single_engine, axis=0) #for循环语句中每次循环都产生一个数据块
train_batch = np.vstack([train_batch, single_engine]) #利用np.vstack将for循环产生的数据块填充至上面建立的空的np.array中
总的来说,在我们需要将for语句产生的数据逐次填充到一个数组当中时,就可以使用np.vstack。