从 4 个 numpy 数组 Python 将坐标写入文件

Question

有 4 个 numpy 矩阵，例如，3x3 坐标：

Xg [[-0.5  0.3  1.1]
    [-0.5  0.3  1.1]
    [-0.5  0.3  1.1]]

Yg [[-0.5 -0.5 -0.5]
    [ 0.3  0.3  0.3]
    [ 1.1  1.1  1.1]]

 u [[ 1.  1.  1.]
    [ 1.  1.  1.]
    [ 1.  1.  1.]]

 v [[ 1.03793  0.25065 -0.28944]
    [-0.21591 -0.93072 -0.10047]
    [-0.08591 -0.11284 -0.06082]]

如何在文件中写入坐标：

# in file should be ", {{" x_coordinate","y_coordinate"},{"u_coordinate","v_coordinate"}}")
 file = open("coordinates.txt", "w")
 file.write(",{{" + x + "," + y + "},{" + u + "," + v + "}}")

 #Output
  ,{{-0.5,-0.5},{1,1.03793}}, {{0.3,-0.5},{1,0.25065}}, {{1.1,-0.5},{1,-0.28944}},...

Answer 1

你可以做嵌套的 for 循环，像这样：

 X = [[-0.5, 0.3, 1.1],
      [-0.5, 0.3, 1.1],
      [-0.5, 0.3, 1.1]]

Y = [[-0.5, -0.5, -0.5],
      [0.3, 0.3, 0.3],
      [1.1, 1.1, 1.1]]

U = [[1, 1, 1, ],
     [1, 1, 1, ],
     [1, 1, 1, ]]

V = [[1.03793, 0.25065, -0.28944],
     [-0.21591, -0.93072, -0.10047],
     [-0.08591, -0.11284, -0.06082]]

with open("coordinates.txt", "w") as f:
    for i in range(3):
        for j in range(3):
            f.write("{{{0},{1}}}, {{{2}, {3}}}\n".format(X[j][i], Y[j][i],  U[j][i], V[j][i]))

这给了

{-0.5,-0.5}, {1, 1.03793}
{-0.5,0.3}, {1, -0.21591}
{-0.5,1.1}, {1, -0.08591}
{0.3,-0.5}, {1, 0.25065}
{0.3,0.3}, {1, -0.93072}
{0.3,1.1}, {1, -0.11284}
{1.1,-0.5}, {1, -0.28944}
{1.1,0.3}, {1, -0.10047}
{1.1,1.1}, {1, -0.06082}