排序XYZ座標數組到表中的Python

Question

我很新的蟒蛇，我無法弄清楚如何來解決這個問題。 我有一個有很多X，Y，Z座標的數組，我需要創建一個輸出文件將它們導入exel。

我導入的STL文件，並把創建從該文件中的二維數組

眼下陣列是這樣的：

example = [[X1,Y1,Z1],[X1,Y2,Z2],[X2,Y1,Z3],[X2,Y2,Z4]] 

的X和Y座標重複自己很多和Z軸始終是不同。

我需要做的就是給他們用這個佈局排序，將它保存到一個.csv文件：

example, Y1, Y2 
X1, Z1, Z2 
X2, Z3, Z4 

所以就把X座標爲行，該Y列，並在其相應的位置在Z 這可以幫助我嗎？ 非常感謝。

Answer 1

你可以打破這個問題分成以下幾個步驟：

1. 獲取所有的獨特的X和Y座標
2. 構建適當大小的表/矩陣
3. 分配x和y座標沿頂邊緣和左邊緣
4. 迭代通過陣列，抓住z座標，並根據在矩陣其映射到正確的位置其x和y座標
5. 輸出所得到的矩陣作爲csv文件。

我做如下假設：

1. 如果給定x, y, z座標陣列中不存在，但在基質中爲它提供一個空間，在矩陣中的對應點會具有值「0」
2. 數組中沒有重複的座標。

鑑於這些假設，下面的程序應該大致做我認爲你想要的。

def find_x_and_y(array): 
    '''Step 1: Get unique x and y coordinates, 
    and the width and height of the matrix''' 
    x = sorted(list(set([i[0] for i in array]))) 
    y = sorted(list(set([i[1] for i in array]))) 

    width = len(x) + 1 
    height = len(y) + 1 

    return x, y, width, height 

def construct_initial_matrix(array): 
    '''Step 2: Make the initial matrix (filled with zeros)''' 
    x, y, width, height = find_x_and_y(array) 

    matrix = [] 
    for i in range(height): 
     matrix.append([0] * width) 

    return matrix 

def add_edging(array, matrix): 
    '''Step 3: Add the x and y coordinates to the edges''' 
    x, y, width, height = find_x_and_y(array) 

    for coord, position in zip(x, range(1, height)): 
     matrix[position][0] = coord 

    for coord, position in zip(y, range(1, width)): 
     matrix[0][position] = coord 

    return matrix 

def add_z_coordinates(array, matrix): 
    '''Step 4: Map the coordinates in the array to the position 
    in the matrix''' 
    x, y, width, height = find_x_and_y(array) 

    x_to_pos = dict(zip(x, range(1, height))) 
    y_to_pos = dict(zip(y, range(1, width))) 

    for x, y, z in array: 
     matrix[x_to_pos[x]][y_to_pos[y]] = z 
    return matrix 

def make_csv(matrix): 
    '''Step 5: Pretty-printing''' 
    return '\n'.join(', '.join(str(i) for i in row) for row in matrix) 

def main(): 
    #example = [[1, 1, 10], [1, 2, 11], [2, 1, 12], [2, 2, 13]] 
    example = [[1000,250,12.2],[1000,500,10],[2000,250,15],[2000,500,13.5]] 

    matrix = construct_initial_matrix(example) 
    matrix = add_edging(example, matrix) 
    matrix = add_z_coordinates(example, matrix) 

    print make_csv(matrix) 

main()