remplissage d'un array 3D

Bonjour,
je cale sur ce truc : il faut que je transforme cet array RAW en un array 3d comportant les elements z aux coordonnées x et y.

donc a partir de :
Code:

1 2 3 4 5 6 7 8 9 10 RAW = [\ [1,1,10],\ [1,2,20],\ [2,1,30],\ [2,2,40],\ [1,1,110],\ [1,2,120],\ [2,1,130],\ [2,2,140],\ ]
il me faudrait
Code:

1 2 3 4 5 6 7 8 9 10 11 X = [1,2] Y = [1,2] Z = [ [[10,110], [20,120]], [[30,130], [40,140]], ]
J'ai essayé avec des append, des insert, des concatenate, mais ca ne le fait pas :
Code:

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 #!/usr/bin/python import os import sys from pylab import * import numpy as np import time import datetime RAW = [\ [1,1,10],\ [1,2,20],\ [2,1,30],\ [2,2,40],\ [1,1,110],\ [1,2,120],\ [2,1,130],\ [2,2,140],\ ] RAW = np.transpose(RAW) X = np.unique(RAW[0]) Y = np.unique(RAW[1]) Z = np.zeros((len(X),len(Y),1)) print Z RAW = np.transpose(RAW) for raw in RAW: x = raw[0] y = raw[1] z = raw[2] iX = np.where(X==x)[0][0] iY = np.where(Y==y)[0][0] print Z[iX,iY] print np.append(Z[iX,iY],z) Z[iX,iY] = np.append(Z[iX,iY],z)
ça par exemple, ça me retourne : "ValueError: output operand requires a reduction, but reduction is not enabled"

c'est un exemple simplifié, car a la base, mon tableau RAW n'est pas nécessairement trié (et beaucoup plus gros).

Merci pour votre aide !

Je propose ceci:

Code:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
import numpy as np
 
RAW = [[1,1,10],
       [1,2,20],
       [2,1,30],
       [2,2,40],
       [1,1,110],
       [1,2,120],
       [2,1,130],
       [2,2,140],
       ]
 
RAW = np.array(RAW)
 
X = np.unique(RAW[:,0])
Y = np.unique(RAW[:,1])
 
Z = [[[] for _ in Y] for _ in X]
for (x,y,z) in RAW:
    Z[where(X==x)[0]][where(Y==y)[0]].append(z)
 
Z = np.array(Z)

merci !

je n'aurai pas pensé qu'il fut nécessaire de repasser par une liste...
Mais ça marche bien, merci !