我有一个数组A.我想拿第一个,第二个,...,每个A[0],A[1],A[2]的第九个元素形成新数组B.我给出了当前和预期输出.

import numpy as np

A=np.array([np.array([[1, 2, 3],
              [4, 5, 6 ],
              [7, 8, 9]]),
       np.array([[[10, 11, 12],
               [13, 14, 15 ],
               [16, 17, 18]]]),
       np.array([[[19, 20, 21],
               [22, 23, 24],
               [25, 26, 27]]])], dtype=object)

for t in range(0,len(A)):
    B=A[0][t][0]
    print([B])

电流输出为

[1]
[4]
[7]

预期输出为

array([[1,10,19],
        [2,11,20],
        [3,12,21],
        [4,13,22],
        [5,14,23],
        [6,15,24],
        [7,16,25],
        [8,17,26],
        [9,18,27]])

推荐答案

您可以遍历数组,将所有值作为列附加,并转置生成的矩阵:

import numpy as np

A=np.array([np.array([[1, 2, 3],
              [4, 5, 6 ],
              [7, 8, 9]]),
       np.array([[[10, 11, 12],
               [13, 14, 15 ],
               [16, 17, 18]]]),
       np.array([[[19, 20, 21],
               [22, 23, 24],
               [25, 26, 27]]])], dtype=object)
               
out = np.array([A[i].flatten() for i in range(len(A))]).transpose()
#out = np.array([i.flatten() for i in A]).transpose() #Second option
print(out)

Output:

[[ 1 10 19]
 [ 2 11 20]
 [ 3 12 21]
 [ 4 13 22]
 [ 5 14 23]
 [ 6 15 24]
 [ 7 16 25]
 [ 8 17 26]
 [ 9 18 27]]

Python相关问答推荐

从numpy数组和参数创建收件箱

索引到 torch 张量,沿轴具有可变长度索引

需要帮助重新调整python fill_between与数据点

在pandas/python中计数嵌套类别

Pandas:计算中间时间条目的总时间增量

合并与拼接并举

在代码执行后关闭ChromeDriver窗口

Gunicorn无法启动Flask应用,因为无法将应用解析为属性名或函数调用.'"'' "

并行编程:同步进程

仅使用预先计算的排序获取排序元素

需要帮助使用Python中的Google的People API更新联系人的多个字段'

在Django中重命名我的表后,旧表中的项目不会被移动或删除

多个矩阵的张量积

使用np.fft.fft2和cv2.dft重现相位谱.为什么结果并不相似呢?

将像素信息写入文件并读取该文件

与同步和异步客户端兼容的Python函数

在不中断格式的情况下在文件的特定部分插入XML标签

使用Scikit的ValueError-了解

按最大属性值Django对对象进行排序

在Django REST框架中定义的URL获得404分