有没有办法在不使用for循环的情况下实现与以下代码相同的输出?
A:
[[[ 1 2 3]
[ 4 5 6]]
[[ 7 8 9]
[10 11 12]]]
输出:
[[ 0 1 2 3 4 5 6 7 8 9 10 11]
[12 13 14 15 16 17 18 19 20 21 22 23]]
代码:
A=np.arange(24).reshape(2,3,4)
v= []
for i in range(A.shape[0]):
v.append(np.concatenate(A[i]))
答案 0 :(得分:2)
再做一次重塑!
v = A.reshape(A.shape[0], -1)
将来使用Python REPL进行实验可能会有所帮助。