欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

两个二维数组一一对应合成一个三维数组

程序员文章站 2022-03-03 10:11:36
...

在改代码的时候遇到的关于数据形式的问题。

需要的数据类型是:

pat = [
        [[0,0], [0]],
        [[0,1], [1]],
        [[1,0], [1]],
        [[1,1], [0]]
    ]

但是因为我的数据是从excel表格导入的,所以要么是一整个数组要么是两个数组。

就只能先分组读入:

a=[
   [0,0],
   [0,1],
   [1,0],
   [1,1]

]

b=[
   [0],
   [1],
   [1],
   [0]

]

然后通过循环合成对应的三维数组:

import os
import numpy as np
a=[
   [0,0],
   [0,1],
   [1,0],
   [1,1]

]
b=[
   [0],
   [1],
   [1],
   [0]]

for i in range(4):
    a[i]=[a[i],b[i]]
print (a)

 

相关标签: Python 数据处理