如何使用for回圈使用陣列的所有元素？-有解無憂

實際上，我需要將函式 (global_displacement(X)) 的回傳值放入另一個運行回圈中。

有人可以告訴我如何獲得所需的輸出嗎？

以及我犯了什么愚蠢的錯誤。

因為每次它只給我第一個([ 0, 0, X[0], X[1]])或

輸出中的最后一個值（[ X[20], X[21], X[53], X[54]]），

因為下面撰寫的代碼中“return j”的縮進錯誤。

import numpy as np 


X = [ 0.19515612,  0.36477665,  0.244737,    0.42873321, 0.16864666,  0.08636661,  0.05376605, -0.57201897, -0.00935055, -1.24923862,  0.,         -1.53111525,  0.00935055, -1.24923862, -0.05376605, -0.57201897, -0.1686466,
     0.08636661, -0.244737,    0.42873321, -0.19515612,  0.36477665,  0.02279911,  0.  ,        0.3563355 ,  0.01379104,  0.   ,       0.42289958, -0.00747999,  0.   ,       0.0825908,  -0.02949519 , 0.   ,      -0.57435396,
     -0.04074819,  0.   ,      -1.25069528 ,-0.02972642,  0.    ,     -1.53227704, -0.    ,      0.   ,      -1.25069528 , 0.02972642 , 0.   ,      -0.57435396 , 0.04074819 , 0.     ,     0.0825908,   0.02949519,  0.  ,
        0.42289958,  0.00747999 , 0.      ,    0.3563355 , -0.01379104, -0.02279911]



def global_displacement(X):
    
    global_displacements = np.array( [[ 0,  0, X[0],  X[1]], [ X[0], X[1], X[2], X[3]], [ X[2],  X[3],X[4],  X[5]], [ X[4],X[5],X[6], X[7]],[ X[6],X[7],X[8],X[9]], [ X[8],X[9],X[10], X[11] ], [ X[10], X[11],X[12], X[13]], [ X[12], X[13],X[14], X[15]],[ X[14], X[15],X[16], X[17]],[ X[16], X[17],X[18], X[19]], [ X[18], X[19],X[20], X[21]],[ X[20], X[21], 0, 0],
                            [ X[0], X[1], X[23], X[24]], [ X[2], X[3], X[26],X[27]], [ X[4], X[5], X[29],X[30]], [ X[6], X[7], X[32],X[33]], [ X[8],X[9],X[35], X[36]], [ X[10], X[11], X[38], X[39]], [ X[12], X[13], X[41], X[42]] ,[ X[14], X[15], X[44], X[45]],[ X[16], X[17], X[47], X[48]],[ X[18], X[19], X[50], X[51]], [ X[20], X[21], X[53], X[54]] ] )
    
    for i in (global_displacements):
        j =  i.reshape(4,1)
        return j

print(global_displacement(X))

這是預期的輸出，我需要通過呼叫此函式將這些值放入另一個回圈中。

[[0.        ]
 [0.        ]
 [0.19515612]
 [0.36477665]]
[[0.19515612]
 [0.36477665]
 [0.244737  ]
 [0.42873321]]
[[0.244737  ]
 [0.42873321]
 [0.16864666]
 [0.08636661]]
[[ 0.16864666]
 [ 0.08636661]
 [ 0.05376605]
 [-0.57201897]]
[[ 0.05376605]
 [-0.57201897]
 [-0.00935055]
 [-1.24923862]]
[[-0.00935055]
 [-1.24923862]
 [ 0.        ]
 [-1.53111525]]
[[ 0.        ]
 [-1.53111525]
 [ 0.00935055]
 [-1.24923862]]
[[ 0.00935055]
 [-1.24923862]
 [-0.05376605]
 [-0.57201897]]
[[-0.05376605]
 [-0.57201897]
 [-0.1686466 ]
 [ 0.08636661]]
[[-0.1686466 ]
 [ 0.08636661]
 [-0.244737  ]
 [ 0.42873321]]
[[-0.244737  ]
 [ 0.42873321]
 [-0.19515612]
 [ 0.36477665]]
[[-0.19515612]
 [ 0.36477665]
 [ 0.        ]
 [ 0.        ]]
[[0.19515612]
 [0.36477665]
 [0.        ]
 [0.3563355 ]]
[[0.244737  ]
 [0.42873321]
 [0.        ]
 [0.42289958]]
[[0.16864666]
 [0.08636661]
 [0.        ]
 [0.0825908 ]]
[[ 0.05376605]
 [-0.57201897]
 [ 0.        ]
 [-0.57435396]]
[[-0.00935055]
 [-1.24923862]
 [ 0.        ]
 [-1.25069528]]
[[ 0.        ]
 [-1.53111525]
 [ 0.        ]
 [-1.53227704]]
[[ 0.00935055]
 [-1.24923862]
 [ 0.        ]
 [-1.25069528]]
[[-0.05376605]
 [-0.57201897]
 [ 0.        ]
 [-0.57435396]]
[[-0.1686466 ]
 [ 0.08636661]
 [ 0.        ]
 [ 0.0825908 ]]
[[-0.244737  ]
 [ 0.42873321]
 [ 0.        ]
 [ 0.42289958]]
[[-0.19515612]
 [ 0.36477665]
 [ 0.        ]
 [ 0.3563355 ]]

uj5u.com熱心網友回復：

您的函式已經將所有內容轉換為正確的格式，除了內部值應存盤到串列中。為此，您可以使用numpy.newaxis. 它用于為您的陣列添加一個新維度（關于其功能的好帖子）。

import numpy as np

def global_displacement(X):
    global_displacements = np.array( [[ 0,  0, X[0],  X[1]], [ X[0], X[1], X[2], X[3]], [ X[2],  X[3],X[4],  X[5]], [ X[4],X[5],X[6], X[7]],[ X[6],X[7],X[8],X[9]], [ X[8],X[9],X[10], X[11] ], [ X[10], X[11],X[12], X[13]], [ X[12], X[13],X[14], X[15]],[ X[14], X[15],X[16], X[17]],[ X[16], X[17],X[18], X[19]], [ X[18], X[19],X[20], X[21]],[ X[20], X[21], 0, 0],
                                [ X[0], X[1], X[23], X[24]], [ X[2], X[3], X[26],X[27]], [ X[4], X[5], X[29],X[30]], [ X[6], X[7], X[32],X[33]], [ X[8],X[9],X[35], X[36]], [ X[10], X[11], X[38], X[39]], [ X[12], X[13], X[41], X[42]] ,[ X[14], X[15], X[44], X[45]],[ X[16], X[17], X[47], X[48]],[ X[18], X[19], X[50], X[51]], [ X[20], X[21], X[53], X[54]] ] )

    new_structure = global_displacements[:, :, np.newaxis]   
    return new_structure

X = [ 0.19515612,  0.36477665,  0.244737,    0.42873321, 0.16864666,  0.08636661,  0.05376605, -0.57201897, -0.00935055, -1.24923862,  0.,         -1.53111525,  0.00935055, -1.24923862, -0.05376605, -0.57201897, -0.1686466,
     0.08636661, -0.244737,    0.42873321, -0.19515612,  0.36477665,  0.02279911,  0.  ,        0.3563355 ,  0.01379104,  0.   ,       0.42289958, -0.00747999,  0.   ,       0.0825908,  -0.02949519 , 0.   ,      -0.57435396,
     -0.04074819,  0.   ,      -1.25069528 ,-0.02972642,  0.    ,     -1.53227704, -0.    ,      0.   ,      -1.25069528 , 0.02972642 , 0.   ,      -0.57435396 , 0.04074819 , 0.     ,     0.0825908,   0.02949519,  0.  ,
        0.42289958,  0.00747999 , 0.      ,    0.3563355 , -0.01379104, -0.02279911]

result = global_displacement(X)

print(result)

輸出：

[[[ 0.        ]
  [ 0.        ]
  [ 0.19515612]
  [ 0.36477665]]

 [[ 0.19515612]
  [ 0.36477665]
  [ 0.244737  ]
  [ 0.42873321]]

 [[ 0.244737  ]
  [ 0.42873321]
  [ 0.16864666]
  [ 0.08636661]]

 [[ 0.16864666]
  [ 0.08636661]
  [ 0.05376605]
  [-0.57201897]]

 [[ 0.05376605]
  [-0.57201897]
  [-0.00935055]
  [-1.24923862]]

 [[-0.00935055]
  [-1.24923862]
  [ 0.        ]
  [-1.53111525]]

 [[ 0.        ]
  [-1.53111525]
  [ 0.00935055]
  [-1.24923862]]

 [[ 0.00935055]
  [-1.24923862]
  [-0.05376605]
  [-0.57201897]]

 [[-0.05376605]
  [-0.57201897]
  [-0.1686466 ]
  [ 0.08636661]]

 [[-0.1686466 ]
  [ 0.08636661]
  [-0.244737  ]
  [ 0.42873321]]

 [[-0.244737  ]
  [ 0.42873321]
  [-0.19515612]
  [ 0.36477665]]

 [[-0.19515612]
  [ 0.36477665]
  [ 0.        ]
  [ 0.        ]]

 [[ 0.19515612]
  [ 0.36477665]
  [ 0.        ]
  [ 0.3563355 ]]

 [[ 0.244737  ]
  [ 0.42873321]
  [ 0.        ]
  [ 0.42289958]]

 [[ 0.16864666]
  [ 0.08636661]
  [ 0.        ]
  [ 0.0825908 ]]

 [[ 0.05376605]
  [-0.57201897]
  [ 0.        ]
  [-0.57435396]]

 [[-0.00935055]
  [-1.24923862]
  [ 0.        ]
  [-1.25069528]]

 [[ 0.        ]
  [-1.53111525]
  [ 0.        ]
  [-1.53227704]]

 [[ 0.00935055]
  [-1.24923862]
  [ 0.        ]
  [-1.25069528]]

 [[-0.05376605]
  [-0.57201897]
  [ 0.        ]
  [-0.57435396]]

 [[-0.1686466 ]
  [ 0.08636661]
  [ 0.        ]
  [ 0.0825908 ]]

 [[-0.244737  ]
  [ 0.42873321]
  [ 0.        ]
  [ 0.42289958]]

 [[-0.19515612]
  [ 0.36477665]
  [ 0.        ]
  [ 0.3563355 ]]]

uj5u.com熱心網友回復：

首先，您不需要.reshape將包含 N 個元素的 1D 陣列轉換為 N×1 的 2D 陣列。您只需向陣列添加一個維度即可。

其次，您通常不想撰寫回圈來處理 Numpy 陣列。您想使用 Numpy 工具一次處理所有內容。只需考慮全部維度的問題：您想將 M x N 的 2D 陣列轉換為 M x N x 1 的 3D 陣列。那......仍然只是向陣列添加一個維度。

所以：

global_displacements = np.array(...)
return global_displacements[..., np.newaxis]

轉載請註明出處，本文鏈接：https://www.uj5u.com/net/418747.html

標籤：

上一篇：For回圈遍歷用戶輸入以計算成本？

下一篇：訪問嵌套的JSON物件而不在JavaScript中指定鍵