python - 在行和列中拆分数组

Question

使用 Python 2。

我需要将一个数组拆分成它们的行和列，但我似乎没有得到练习中要求的解决方案

import numpy as np
a = np.array([[5, 0, 3, 3],
 [7, 9, 3, 5],
 [2, 4, 7, 6],
 [8, 8, 1, 6]])

到目前为止我有这些功能

def _rows(a):
    print("array:"+ str(a[:,]))
_rows(a)
def _col(a):
    alt=a.T
    print ("array:"+ str(alt[:,]))
_col(a)

但我需要返回一个列表，当我使用 list() 函数时，它会将每个单独的字符分开

我需要的结果是:

[array([5, 0, 3, 3]), array([7, 9, 3, 5]), array([2, 4, 7, 6]), array([8, 8, 1, 6])]

[array([5, 7, 2, 8]), array([0, 9, 4, 8]), array([3, 3, 7, 1]), array([3, 5, 6, 6])]

Answer 1

您可以将行和列解压缩到一个列表中:

res1, res2 = [*a], [*a.T]

---

print(res1)

[array([5, 0, 3, 3]),
 array([7, 9, 3, 5]),
 array([2, 4, 7, 6]),
 array([8, 8, 1, 6])]

print(res2)

[array([5, 7, 2, 8]),
 array([0, 9, 4, 8]),
 array([3, 3, 7, 1]),
 array([3, 5, 6, 6])]

---

Extended iterable unpacking是在 python 3.0 中引入的，对于旧版本，您可以调用列表构造函数，如@U9-Forward 的回答