gpt4 book ai didi

python - 如何将 3D 数组转换为 python 中的 2D 数组列表?

转载 作者:行者123 更新时间:2023-12-05 09:34:38 25 4
gpt4 key购买 nike

我想将 3D 数组(比如大小 = 3x3x4)转换为 3 (3x4) 个数组的列表。else block 中的 3D 数组 Aif block 中的类型不同。

我试过 tolist() 函数,但它会将 3D 数组转换为列表的列表,这是不需要的。

X = np.random.randn(3, 3, 3)
for R in range(1,10):
N = len(X.shape)
B = ()
C = []

for i in range(N):
B += svd(X[i])
C.append(B[3*i])

C = np.array(C)
if R <= np.shape(C)[2]:
A = C[:,:,:R]
else:
A = np.concatenate((C[:,:,:np.shape(C)[2]],np.zeros((3,3,1))),axis=2)

理想情况下,A = [np.random.randn(In, R) for In in X.shape]。但我通过将 X 的左奇异向量分配给 A 来修改此步骤。

最佳答案

numpy 数组是可迭代的,您可以轻松地将其转换为列表:

lst = list(A)

演示:

>>> arr = np.arange(8).reshape(2,2,2)
>>> arr
array([[[0, 1],
[2, 3]],

[[4, 5],
[6, 7]]])
>>> list(arr)
[array([[0, 1],
[2, 3]]), array([[4, 5],
[6, 7]])]

关于python - 如何将 3D 数组转换为 python 中的 2D 数组列表?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/66495623/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com