Dimension nicht für eine nd Array passende

Question

Ich habe eine nd Array erstellt aus einer Datei wie:

for i in range(int(atoms)): 
    next(finp) 
    for j in range(int(ldata[2])): 
     aatom[i][j] = [float(x) for x in 
         finp.readline().strip().split()] 

Welche ich erwarte einen 3D-Array zu sein (i,j,x). Aber es ist nicht. Nachdem es zu einem Array numpy Umwandlung als:

atom = np.array(aatom) 
print(atom.shape) 

yeilds: (250, 301), die die Dimension der i,j ist. Aber in meinem denken Prozess wird es in der Tat 3d als:

print(atom[1][1][:]) 

yeilds: [-3.995, 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]

Was ich hier falsch mache?

Answer 1

Ihre verschachtelten Listen haben wahrscheinlich unterschiedliche Längen, wodurch NumPy gezwungen ist, ein Array von dtype=object zu erstellen, was ein Array von list s anstelle von float s bedeutet.

vergleichen

a = [range(3), range(2)] # different lenghts in inner lists 
aa = np.array(a) 
print aa 
print aa.shape   # (2,) 
print aa.dtype   # dtype('O') 

und

b = [range(3), range(3)] # equal lenghts in inner lists 
bb = np.array(b) 
print bb 
print bb.shape   # (2, 3) 
print bb.dtype   # dtype('int64')