python - 在 Python 中重新格式化表

Question

我在 Python 脚本中有一个带有 numpy 的表，其形状如下:

[array([[a1, b1, c1], ..., [x1, y1, z1]]), 
 array([a2, b2, c2, ..., x2, y2, z2])
]

我想将其 reshape 为这样的格式:

(array([[a2],
        [b2],
         .
         .
         .
        [z2]],
        dtype = ...),
 array([[a1],
        [b1],
         .
         .
         .
        [z1]])
)

说实话，我对不同的括号也很困惑。 array1, array2] 是一个数组列表，对吗？那么(array1, array2) 是什么？

Answer 1

圆括号(1, 2)为 tuples , 方括号 [1, 2] 为 lists 。要转换数据结构，请使用 expand_dims和 flatten .

import numpy as np

a = [
    np.array([[1, 2, 3], [4, 5, 6]]),
    np.array([10, 11, 12, 13, 14])
    ]
print(a)

b = (
    np.expand_dims(a[1], axis=1),
    np.expand_dims(a[0].flatten(), axis=1)
    )

print(b)