将nan添加到列表中

Question

我正在尝试将nan值添加到mem_labels1中，以便它可以具有与映像1相同数量的值。我原以为这段代码可以工作，但它仍然在每个代码中保留不同数量的值。我该怎么修？

for id_code in list_of_done_files:
if id_code in id_mem_dict:
mem_labels1.append(id_mem_dict[id_code])
else:
mem_labels1.append(np.nan)

images1 = np.array(images1, dtype=object)
mem_labels1 = np.fromiter(id_mem_dict.values(), dtype=float)
mem_labels1 = np.array(mem_labels1)

Answer 1

@mozway的答案经过一些调整后有效。

mem_labels1 = np.pad(
mem_labels1.astype(float), (0, len(images1) - len(mem_labels1)),
'constant', constant_values=(np.nan, np.nan)
)

Answer 2

您可以pad您的数组：

mem_labels1 = np.pad(mem_labels1.astype(float),
(0, len(mem_labels1)-len(images1)),
'constant',
constant_values=(np.nan, np.nan))