numpy索引列表错误

Question

我正在尝试创建一个索引的np.ndarray：

dos2[q] = np.empty((dos[q].shape[0], dos[q].shape[1] + 1), dtype=dos[q].dtype)

正确定义q。 我收到了错误：

NameError: name 'dos2' is not defined

dos [q]的形状和类型是：

<class 'numpy.ndarray'>
(301, 18)

如果我用q循环创建它：

dos2 = np.empty((dos.shape[0], dos.shape[1] + 1), dtype=dos.dtype)

然后一切都很好。 （在这种情况下，dos也不在q循环中，具有相同的类型和维度）

Answer 1

我想这只是dos2的声明。如果您要定义dos2[q]，则应在

之前定义dos2

Answer 2

当您尝试分配

时，似乎未定义doc2

dos2[q] = np.empty((dos[q].shape[0], dos[q].shape[1] + 1), dtype=dos[q].dtype)

每当您尝试在阵列的一部分中分配某些内容时，您需要先定义它。